| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander_test.cc | 101 ShapeUtil::GetTupleElementShape(while_shape, 1))); in TEST_F()
105 ShapeUtil::GetTupleElementShape(while_shape, 2))); in TEST_F()
109 ShapeUtil::GetTupleElementShape(while_shape, 3))); in TEST_F()
|
| 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 | 163 ShapeUtil::GetTupleElementShape(shaped_buffer->on_host_shape(), i), in DeconstructTuple()
164 ShapeUtil::GetTupleElementShape(shaped_buffer->on_device_shape(), i), in DeconstructTuple()
|
| D | hlo_verifier.cc | 858 const Shape& dest_shape = ShapeUtil::GetTupleElementShape(operand_shape, 0); in HandleCopyDone()
859 const Shape& src_shape = ShapeUtil::GetTupleElementShape(operand_shape, 1); in HandleCopyDone()
869 return CheckShape(copy_done, ShapeUtil::GetTupleElementShape( in HandleCopyDone()
889 {ShapeUtil::GetTupleElementShape(recv->shape(), 0), in HandleRecv()
898 {ShapeUtil::GetTupleElementShape(recv_done->operand(0)->shape(), 0), in HandleRecvDone()
|
| D | tuple_simplifier_test.cc | 157 ShapeUtil::GetTupleElementShape(element->shape(), 0), element, 0)); in TEST_F()
|
| D | hlo_module_group_metadata.cc | 466 ShapeUtil::GetTupleElementShape(channel.recv_done->shape(), 0); in VerifyChannelInstructions()
|
| 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 | bfloat16_propagation_test.cc | 996 EXPECT_EQ(ShapeUtil::GetTupleElementShape(domain->shape(), 0).element_type(), in TEST_F()
998 EXPECT_EQ(ShapeUtil::GetTupleElementShape(domain->shape(), 1).element_type(), in TEST_F()
|
| D | layout_assignment.cc | 413 ShapeUtil::GetTupleElementShape(send_recv_instr->shape(), 0); in BuildHostChannelConstraints()
510 ShapeUtil::GetTupleElementShape(instruction->shape(), 0); in AddMandatoryConstraints()
|
| D | hlo_cost_analysis.cc | 660 ? ShapeUtil::GetTupleElementShape(fft->operand(0)->shape(), 0) in HandleFft()
|
| D | hlo_computation.cc | 743 ShapeUtil::GetTupleElementShape(instruction->shape(), i), in DeepCopyHelper()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_utils.cc | 145 *buffer_shape = xla::ShapeUtil::GetTupleElementShape(list_shape, 0); in GetTensorListBufferShape()
202 auto buffer_shape = xla::ShapeUtil::GetTupleElementShape(list_shape, 0); in GetLeadingDimForTensorList()
217 xla::ShapeUtil::GetTupleElementShape(element_tensor_list_shape, i); in GetTensorListShapeFromElementTensorListShape()
257 xla::ShapeUtil::GetTupleElementShape(list_shape, i); in CreateZerosTensorListWithShape()
323 xla::ShapeUtil::GetTupleElementShape(element_shape, i); in ExecuteTensorListPushBack()
385 xla::ShapeUtil::GetTupleElementShape(list_shape, i); in ExecuteTensorListPopBack()
466 xla::ShapeUtil::GetTupleElementShape(list_shape, 0); in ExecuteTensorListGetItem()
|
| D | stack_ops.cc | 48 return XLAShapeToTensorShape(xla::ShapeUtil::GetTupleElementShape(shape, 0), in GetStackShape()
|
| /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/c/eager/ |
| D | c_api_debug.cc | 107 xla::Shape shape0 = xla::ShapeUtil::GetTupleElementShape(padded_shape, 0); in TensorDebugInfo()
109 xla::ShapeUtil::GetTupleElementShape(padded_shape, 1); in TensorDebugInfo()
|
| /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 | 225 const xla::Shape& result0 = ShapeUtil::GetTupleElementShape(result, 0); in TEST()
268 const xla::Shape& result0 = ShapeUtil::GetTupleElementShape(result, 0); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.cc | 208 ShapeUtil::GetTupleElementShape(literal_shape, i); in TransferLiteralFromOutfeed()
|
| D | ir_emitter.cc | 392 ShapeUtil::GetTupleElementShape(infeed->shape(), 0))); in HandleInfeed()
402 token_slice, ShapeUtil::GetTupleElementShape(infeed->shape(), 1)); in HandleInfeed()
418 ShapeUtil::GetTupleElementShape(data_shape, i); in HandleInfeed()
552 ShapeUtil::GetTupleElementShape(operand_shape, i); in HandleOutfeed()
2522 ShapeUtil::GetTupleElementShape(custom_call->shape(), i); in HandleCustomCall()
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 654 ShapeUtil::GetTupleElementShape(muladd_result, 0); in TEST()
657 ShapeUtil::GetTupleElementShape(muladd_result, 1); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 390 /* 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 | 486 static const Shape& GetTupleElementShape(const Shape& shape, int64 index);
|
| D | literal_test.cc | 1911 ShapeUtil::GetTupleElementShape(Shape(proto.shape()), 0).ToProto(); in TEST_F()
1929 ShapeUtil::GetTupleElementShape(Shape(proto.shape()), 0).ToProto(); in TEST_F()
1934 ShapeUtil::GetTupleElementShape(Shape(proto.shape()), 1).ToProto(); in TEST_F()
|
| D | literal_comparison.cc | 208 total += RecursiveElementCount(ShapeUtil::GetTupleElementShape(shape, i)); in RecursiveElementCount()
|