Searched refs:infeed_instruction_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 227 StatusOr<XlaOp> InfeedWithTokenInternal(const Shape& infeed_instruction_shape,
|
| D | mlir_hlo_builder.cc | 468 const Shape& infeed_instruction_shape, XlaOp token, const string& config) { in InfeedWithTokenInternal() argument
471 infeed_instruction_shape, builder_)); in InfeedWithTokenInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1669 const Shape infeed_instruction_shape = in Infeed() local
1671 *instr.mutable_shape() = infeed_instruction_shape.ToProto(); in Infeed()
1742 const Shape infeed_instruction_shape = in InfeedWithToken() local
1756 return InfeedWithTokenInternal(infeed_instruction_shape, token, config); in InfeedWithToken()
1761 const Shape& infeed_instruction_shape, XlaOp token, const string& config) { in InfeedWithTokenInternal() argument
1763 *instr.mutable_shape() = infeed_instruction_shape.ToProto(); in InfeedWithTokenInternal()
|
| D | xla_builder.h | 639 const Shape& infeed_instruction_shape, XlaOp token, const string& config);
|