Searched refs:XfeedKind (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 547 enum class XfeedKind { enum
554 Status EmitXfeedTransfer(XfeedKind kind, const Shape& shape,
|
| D | ir_emitter.cc | 369 XfeedKind::kInfeed, tuple_element_shape, tuple_element_address)); in HandleInfeed()
378 EmitXfeedTransfer(XfeedKind::kInfeed, data_shape, data_address)); in HandleInfeed()
384 Status IrEmitter::EmitXfeedTransfer(XfeedKind kind, const Shape& shape, in EmitXfeedTransfer()
409 if (kind == XfeedKind::kInfeed) { in EmitXfeedTransfer()
433 if (kind == XfeedKind::kInfeed) { in EmitXfeedTransfer()
458 if (kind == XfeedKind::kInfeed) { in EmitXfeedTransfer()
486 return EmitXfeedTransfer(XfeedKind::kOutfeed, operand_shape, value); in HandleOutfeed()
497 TF_RETURN_IF_ERROR(EmitXfeedTransfer(XfeedKind::kOutfeed, in HandleOutfeed()
|