OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:xla_output_shape
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
while_op.cc
210
<< " -> " << xla::ShapeUtil::HumanString(body.
xla_output_shape
);
212
<< " -> " << xla::ShapeUtil::HumanString(cond.
xla_output_shape
);
221
ctx, xla::ShapeUtil::Compatible(body_input_shape, body.
xla_output_shape
),
225
xla::ShapeUtil::HumanString(body.
xla_output_shape
)));
230
xla::ShapeUtil::Compatible(cond.
xla_output_shape
,
234
xla::ShapeUtil::HumanString(cond.
xla_output_shape
)));
/external/tensorflow/tensorflow/compiler/tf2xla/
xla_compiler.h
212
xla::Shape
xla_output_shape
;
member in struct:tensorflow::XlaCompiler::CompilationResult
xla_compiler.cc
654
result->
xla_output_shape
.Swap(
657
<< xla::ShapeUtil::HumanString(result->
xla_output_shape
);
660
xla::LayoutUtil::SetToDefaultLayout(&result->
xla_output_shape
);
673
xla::ShapeUtil::GetTupleElementShape(result->
xla_output_shape
,
/external/tensorflow/tensorflow/compiler/jit/
xla_compilation_cache.cc
222
build_options.set_result_layout(result.
xla_output_shape
);
Completed in 100 milliseconds