Searched refs:XfeedKind (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 573 enum class XfeedKind { enum
580 Status EmitXfeedTransfer(XfeedKind kind, const Shape& shape,
|
| D | ir_emitter.cc | 433 XfeedKind::kInfeed, tuple_element_shape, tuple_element_address)); in HandleInfeed()
442 EmitXfeedTransfer(XfeedKind::kInfeed, data_shape, data_address)); in HandleInfeed()
448 Status IrEmitter::EmitXfeedTransfer(XfeedKind kind, const Shape& shape, in EmitXfeedTransfer()
467 kind == XfeedKind::kInfeed in EmitXfeedTransfer()
480 if (kind == XfeedKind::kInfeed) { in EmitXfeedTransfer()
504 kind == XfeedKind::kInfeed in EmitXfeedTransfer()
527 return EmitXfeedTransfer(XfeedKind::kOutfeed, operand_shape, value); in HandleOutfeed()
538 TF_RETURN_IF_ERROR(EmitXfeedTransfer(XfeedKind::kOutfeed, in HandleOutfeed()
|