| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_element_type_converter.cc | 70 Shape subshape = ShapeUtil::GetTupleElementShape(shape, i); in GetConvertedTupleShape()
91 const Shape& ele_shape = ShapeUtil::GetTupleElementShape(shape, i); in ConvertTupleElements()
94 const Shape& to_ele_shape = ShapeUtil::GetTupleElementShape(to_shape, i); in ConvertTupleElements()
|
| D | allocation_tracker.cc | 120 ShapeUtil::GetTupleElementShape(shaped_buffer->on_host_shape(), i), in DeconstructTuple()
121 ShapeUtil::GetTupleElementShape(shaped_buffer->on_device_shape(), i), in DeconstructTuple()
|
| D | tuple_simplifier_test.cc | 151 ShapeUtil::GetTupleElementShape(element->shape(), 0), element, 0)); in TEST_F()
|
| D | hlo_cost_analysis.cc | 402 ? ShapeUtil::GetTupleElementShape(fft->operand(0)->shape(), 0) in HandleFft()
|
| D | hlo_computation.cc | 487 ShapeUtil::GetTupleElementShape(instruction->shape(), i), in DeepCopyHelper()
|
| D | layout_assignment_test.cc | 217 ShapeUtil::GetTupleElementShape(tuple->shape(), 1), constant1->shape())); in TEST_F()
|
| D | layout_assignment.cc | 509 ShapeUtil::GetTupleElementShape(instruction->shape(), 0); in AddMandatoryConstraints()
|
| D | hlo_instruction.cc | 454 Shape output_shape = ShapeUtil::GetTupleElementShape(operand->shape(), 0); in CreateRecvDone()
1426 CreateRecv(ShapeUtil::GetTupleElementShape(shape, 0), channel_id()); in CloneWithNewOperands()
|
| D | hlo_evaluator.cc | 2421 ShapeUtil::GetTupleElementShape(operand->shape(), index)); in HandleGetTupleElement()
|
| D | user_computation.cc | 359 Shape element_shape = ShapeUtil::GetTupleElementShape( in AddGetTupleElementInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_test.cc | 101 ShapeUtil::GetTupleElementShape(result->shape(), 0), in XLA_TEST_F()
105 ShapeUtil::GetTupleElementShape(result->shape(), 1), in XLA_TEST_F()
|
| D | literal_test_util.cc | 593 RecursiveElementCount(ShapeUtil::GetTupleElementShape(shape, i)); in RecursiveElementCount()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_transfer_manager.cc | 79 ShapeUtil::GetTupleElementShape(shape, i); in TransferLiteralToInfeed()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function_test.cc | 128 const xla::Shape& result0 = ShapeUtil::GetTupleElementShape(result, 0); in TEST()
|
| D | xla_compiler.cc | 673 xla::ShapeUtil::GetTupleElementShape(result->xla_output_shape, in CompileGraph()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stack_ops.cc | 49 return XLAShapeToTensorShape(xla::ShapeUtil::GetTupleElementShape(shape, 0), in GetStackShape()
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 462 ShapeUtil::GetTupleElementShape(muladd_result, 0); in TEST()
465 ShapeUtil::GetTupleElementShape(muladd_result, 1); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.cc | 210 ShapeUtil::GetTupleElementShape(literal_shape, i); in TransferLiteralFromOutfeed()
|
| D | parallel_cpu_executable.cc | 95 element, ShapeUtil::GetTupleElementShape(shape, i), marked_addresses); in MarkLiveAddressesInOutput()
|
| D | ir_emitter.cc | 353 ShapeUtil::GetTupleElementShape(shape, i); in HandleInfeed()
469 ShapeUtil::GetTupleElementShape(operand_shape, i); in HandleOutfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 416 static const Shape& GetTupleElementShape(const Shape& shape, int64 index);
|
| D | literal_util_test.cc | 1627 ShapeUtil::GetTupleElementShape(proto.shape(), 0); in TEST_F()
1643 ShapeUtil::GetTupleElementShape(proto.shape(), 0); in TEST_F()
1648 ShapeUtil::GetTupleElementShape(proto.shape(), 1); in TEST_F()
|
| D | shape_util.cc | 338 /* static */ const Shape& ShapeUtil::GetTupleElementShape(const Shape& shape, in GetTupleElementShape() function in xla::ShapeUtil
907 ShapeUtil::GetTupleElementShape(shape, i), func, index)); in ForEachSubshapeHelper()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | numpy_bridge.cc | 124 PyShapeInfoFromXlaShape(ShapeUtil::GetTupleElementShape(shape, i))); in PyShapeInfoFromXlaShape()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| D | hlo_parser.cc | 1408 ShapeUtil::GetTupleElementShape(shape, i))) { in ParseTupleLiteral()
|