| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| client_test.cc | 101 ShapeUtil::GetTupleElementShape(result->shape(), 0),
105 ShapeUtil::GetTupleElementShape(result->shape(), 1),
|
| literal_test_util.cc | 593 RecursiveElementCount(ShapeUtil::GetTupleElementShape(shape, i));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_element_type_converter.cc | 70 Shape subshape = ShapeUtil::GetTupleElementShape(shape, i);
91 const Shape& ele_shape = ShapeUtil::GetTupleElementShape(shape, i);
94 const Shape& to_ele_shape = ShapeUtil::GetTupleElementShape(to_shape, i);
|
| allocation_tracker.cc | 120 ShapeUtil::GetTupleElementShape(shaped_buffer->on_host_shape(), i),
121 ShapeUtil::GetTupleElementShape(shaped_buffer->on_device_shape(), i),
|
| tuple_simplifier_test.cc | 151 ShapeUtil::GetTupleElementShape(element->shape(), 0), element, 0));
|
| hlo_cost_analysis.cc | 402 ? ShapeUtil::GetTupleElementShape(fft->operand(0)->shape(), 0)
|
| hlo_computation.cc | 487 ShapeUtil::GetTupleElementShape(instruction->shape(), i),
|
| layout_assignment_test.cc | 217 ShapeUtil::GetTupleElementShape(tuple->shape(), 1), constant1->shape()));
[all...] |
| layout_assignment.cc | 509 ShapeUtil::GetTupleElementShape(instruction->shape(), 0);
[all...] |
| hlo_instruction.cc | 454 Shape output_shape = ShapeUtil::GetTupleElementShape(operand->shape(), 0);
[all...] |
| hlo_evaluator.cc | [all...] |
| user_computation.cc | 359 Shape element_shape = ShapeUtil::GetTupleElementShape(
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_jit_compiled_cpu_function_test.cc | 128 const xla::Shape& result0 = ShapeUtil::GetTupleElementShape(result, 0);
|
| xla_compiler.cc | 673 xla::ShapeUtil::GetTupleElementShape(result->xla_output_shape,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_transfer_manager.cc | 79 ShapeUtil::GetTupleElementShape(shape, i);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| stack_ops.cc | 49 return XLAShapeToTensorShape(xla::ShapeUtil::GetTupleElementShape(shape, 0),
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| tfcompile_test.cc | 462 ShapeUtil::GetTupleElementShape(muladd_result, 0);
465 ShapeUtil::GetTupleElementShape(muladd_result, 1);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_transfer_manager.cc | 210 ShapeUtil::GetTupleElementShape(literal_shape, i);
|
| parallel_cpu_executable.cc | 95 element, ShapeUtil::GetTupleElementShape(shape, i), marked_addresses);
|
| ir_emitter.cc | 353 ShapeUtil::GetTupleElementShape(shape, i);
469 ShapeUtil::GetTupleElementShape(operand_shape, i);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| numpy_bridge.cc | 124 PyShapeInfoFromXlaShape(ShapeUtil::GetTupleElementShape(shape, i)));
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util.h | 416 static const Shape& GetTupleElementShape(const Shape& shape, int64 index);
|
| shape_util.cc | 338 /* static */ const Shape& ShapeUtil::GetTupleElementShape(const Shape& shape,
907 ShapeUtil::GetTupleElementShape(shape, i), func, index));
[all...] |
| literal_util_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser.cc | [all...] |
