Searched refs:tuple_element_shape (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_xfeed.cc | 205 const Shape& tuple_element_shape = ShapeUtil::GetSubshape(shape, {i}); in TransferLiteralToInfeedOnCpu() local
207 tuple_element_shape, sizeof(void*)); in TransferLiteralToInfeedOnCpu()
253 const Shape& tuple_element_shape = in TransferLiteralFromOutfeedOnCpu() local
255 int64_t size = cpu::runtime::GetByteSizeRequirement(tuple_element_shape, in TransferLiteralFromOutfeedOnCpu()
|
| H A D | ir_emitter.cc | 436 const Shape& tuple_element_shape = in HandleInfeed() local
444 EmitBufferPointer(buffer, tuple_element_shape); in HandleInfeed()
447 XfeedKind::kInfeed, tuple_element_shape, tuple_element_address)); in HandleInfeed()
547 const Shape& tuple_element_shape = in HandleOutfeed() local
550 tuple_element_shape, i, MinimumAlignmentForShape(tuple_element_shape), in HandleOutfeed()
553 tuple_element_shape, tuple_element)); in HandleOutfeed()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instructions.h | 1714 for (const Shape& tuple_element_shape : shape().tuple_shapes()) { in output_shapes() local
1715 shapes.push_back(&tuple_element_shape); in output_shapes()
|