OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:argument_shapes
(Results
1 - 6
of
6
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
service.cc
266
tensorflow::gtl::ArraySlice<const Shape*>
argument_shapes
,
272
if (program_shape.parameters_size() !=
argument_shapes
.size()) {
275
argument_shapes
.size());
277
for (int i = 0; i <
argument_shapes
.size(); ++i) {
280
if (!ShapeUtil::Compatible(*
argument_shapes
[i],
287
ShapeUtil::HumanString(*
argument_shapes
[i]).c_str());
291
*
argument_shapes
[i]));
329
std::vector<const Shape*>
argument_shapes
;
local
331
argument_shapes
.push_back(&arg->on_host_shape());
333
return CreateModuleConfig(program_shape,
argument_shapes
, &execution_options
[
all
...]
service.h
278
tensorflow::gtl::ArraySlice<const Shape*>
argument_shapes
,
shape_inference.cc
2423
string
argument_shapes
=
local
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc
258
const std::vector<Shape>&
argument_shapes
,
261
argument_shape_pointers.reserve(
argument_shapes
.size());
262
for (auto& argument_shape :
argument_shapes
) {
local_computation_builder.h
107
const std::vector<Shape>&
argument_shapes
,
xla_client.py
369
def Compile(self,
argument_shapes
=(), compile_options=None, layout_fn=None):
380
argument_shapes
: parameter shapes -- they are first laid out by layout_fn
394
argument_shapes
= [
395
shape.map_leaves(layout_fn) for shape in
argument_shapes
402
self.c_local_computation.Compile(
argument_shapes
, compile_options),
410
argument_shapes
=[Shape.from_numpy(arg) for arg in arguments],
418
argument_shapes
= [Shape.from_numpy(arg) for arg in arguments]
420
argument_shapes
= [
421
shape.map_leaves(layout_fn) for shape in
argument_shapes
424
argument_shapes
= [None for shape in argument_shapes
[
all
...]
Completed in 97 milliseconds