| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander_test.cc | 97 ShapeUtil::GetTupleElementShape(while_shape, 1))); in TEST()
101 ShapeUtil::GetTupleElementShape(while_shape, 2))); in TEST()
105 ShapeUtil::GetTupleElementShape(while_shape, 3))); in TEST()
|
| 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 | hlo_domain_test.cc | 392 ShapeUtil::GetTupleElementShape(infeed_data->shape(), 0), infeed_data, in TEST_F()
399 ShapeUtil::GetTupleElementShape(infeed_data->shape(), 1), infeed_data, in TEST_F()
646 ShapeUtil::GetTupleElementShape(copy1->shape(), 0), copy1, 0)); in TEST_F()
649 ShapeUtil::GetTupleElementShape(tuple0->shape(), 1), tuple0, 1)); in TEST_F()
|
| D | allocation_tracker.cc | 162 ShapeUtil::GetTupleElementShape(shaped_buffer->on_host_shape(), i), in DeconstructTuple()
163 ShapeUtil::GetTupleElementShape(shaped_buffer->on_device_shape(), i), in DeconstructTuple()
|
| D | tuple_simplifier_test.cc | 157 ShapeUtil::GetTupleElementShape(element->shape(), 0), element, 0)); in TEST_F()
|
| D | hlo_module_group_metadata.cc | 448 ShapeUtil::GetTupleElementShape(channel.recv_done->shape(), 0); in VerifyChannelInstructions()
|
| D | hlo_verifier.cc | 709 {ShapeUtil::GetTupleElementShape(recv->shape(), 0), in HandleRecv()
717 {ShapeUtil::GetTupleElementShape(recv_done->operand(0)->shape(), 0), in HandleRecvDone()
|
| D | bfloat16_propagation_test.cc | 966 EXPECT_EQ(ShapeUtil::GetTupleElementShape(domain->shape(), 0).element_type(), in TEST_F()
968 EXPECT_EQ(ShapeUtil::GetTupleElementShape(domain->shape(), 1).element_type(), in TEST_F()
|
| D | layout_assignment_test.cc | 81 const Layout& actual = ShapeUtil::GetTupleElementShape(shape, i).layout(); in ExpectTupleLayoutIs()
246 ShapeUtil::GetTupleElementShape(tuple->shape(), 1), constant1->shape())); in TEST_F()
|
| D | hlo_cost_analysis.cc | 538 ? ShapeUtil::GetTupleElementShape(fft->operand(0)->shape(), 0) in HandleFft()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_debug.cc | 99 xla::Shape shape0 = xla::ShapeUtil::GetTupleElementShape(padded_shape, 0); in TFE_TensorHandleTensorDebugInfo()
101 xla::ShapeUtil::GetTupleElementShape(padded_shape, 1); in TFE_TensorHandleTensorDebugInfo()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_test.cc | 103 ShapeUtil::GetTupleElementShape(result.shape(), 0), in XLA_TEST_F()
107 ShapeUtil::GetTupleElementShape(result.shape(), 1), in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.cc | 34 xla::ShapeUtil::GetTupleElementShape(shape, i); in PopulateInfeedLayoutVector()
141 xla::ShapeUtil::GetTupleElementShape(input_shape, i); in GetShapeWithLayout()
|
| D | xla_jit_compiled_cpu_function_test.cc | 128 const xla::Shape& result0 = ShapeUtil::GetTupleElementShape(result, 0); in TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_utils.cc | 63 xla::ShapeUtil::GetTupleElementShape(list_shape, 0), buffer_shape)); in GetTensorListBufferShape()
|
| D | stack_ops.cc | 48 return XLAShapeToTensorShape(xla::ShapeUtil::GetTupleElementShape(shape, 0), in GetStackShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.cc | 208 ShapeUtil::GetTupleElementShape(literal_shape, i); in TransferLiteralFromOutfeed()
|
| D | ir_emitter.cc | 333 ShapeUtil::GetTupleElementShape(infeed->shape(), 0))); in HandleInfeed()
343 token_slice, ShapeUtil::GetTupleElementShape(infeed->shape(), 1)); in HandleInfeed()
359 ShapeUtil::GetTupleElementShape(data_shape, i); in HandleInfeed()
493 ShapeUtil::GetTupleElementShape(operand_shape, i); in HandleOutfeed()
2279 ShapeUtil::GetTupleElementShape(custom_call->shape(), i); in HandleCustomCall()
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 586 ShapeUtil::GetTupleElementShape(muladd_result, 0); in TEST()
589 ShapeUtil::GetTupleElementShape(muladd_result, 1); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xrt.cc | 289 ShapeUtil::GetTupleElementShape(tuple_shape, i); in DestructureXrtAllocationTuple()
|
| D | numpy_bridge.cc | 141 PyShapeInfoFromXlaShape(ShapeUtil::GetTupleElementShape(shape, i)) in PyShapeInfoFromXlaShape()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 343 /* static */ const Shape& ShapeUtil::GetTupleElementShape(const Shape& shape, in GetTupleElementShape() function in xla::ShapeUtil
866 ShapeUtil::GetTupleElementShape(shape, i), func, index)); in ForEachSubshapeHelper()
|
| D | shape_util.h | 473 static const Shape& GetTupleElementShape(const Shape& shape, int64 index);
|
| D | literal_test.cc | 1924 ShapeUtil::GetTupleElementShape(Shape(proto.shape()), 0).ToProto(); in TEST_F()
1942 ShapeUtil::GetTupleElementShape(Shape(proto.shape()), 0).ToProto(); in TEST_F()
1947 ShapeUtil::GetTupleElementShape(Shape(proto.shape()), 1).ToProto(); in TEST_F()
|
| D | literal_comparison.cc | 193 total += RecursiveElementCount(ShapeUtil::GetTupleElementShape(shape, i)); in RecursiveElementCount()
|