HomeSort by relevance Sort by last modified time
    Searched refs:XfeedKind (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.h 547 enum class XfeedKind {
554 Status EmitXfeedTransfer(XfeedKind kind, const Shape& shape,
ir_emitter.cc 369 XfeedKind::kInfeed, tuple_element_shape, tuple_element_address));
378 EmitXfeedTransfer(XfeedKind::kInfeed, data_shape, data_address));
384 Status IrEmitter::EmitXfeedTransfer(XfeedKind kind, const Shape& shape,
409 if (kind == XfeedKind::kInfeed) {
433 if (kind == XfeedKind::kInfeed) {
458 if (kind == XfeedKind::kInfeed) {
486 return EmitXfeedTransfer(XfeedKind::kOutfeed, operand_shape, value);
497 TF_RETURN_IF_ERROR(EmitXfeedTransfer(XfeedKind::kOutfeed,
    [all...]

Completed in 1132 milliseconds