Home
last modified time | relevance | path

Searched defs:tuple_element_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_xfeed.cc205 const Shape& tuple_element_shape = ShapeUtil::GetSubshape(shape, {i}); in TransferLiteralToInfeedOnCpu() local
253 const Shape& tuple_element_shape = in TransferLiteralFromOutfeedOnCpu() local
H A Dir_emitter.cc436 const Shape& tuple_element_shape = in HandleInfeed() local
547 const Shape& tuple_element_shape = in HandleOutfeed() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instructions.h1714 for (const Shape& tuple_element_shape : shape().tuple_shapes()) { in output_shapes() local