Home
last modified time | relevance | path

Searched defs:infeed_instruction_shape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
H A Dmlir_hlo_builder.cc598 const Shape& infeed_instruction_shape, XlaOp token, in InfeedWithTokenInternal()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc1754 const Shape infeed_instruction_shape = in Infeed() local
1827 const Shape infeed_instruction_shape = in InfeedWithToken() local
1846 const Shape& infeed_instruction_shape, XlaOp token, in InfeedWithTokenInternal()