HomeSort by relevance Sort by last modified time
    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