Home
last modified time | relevance | path

Searched refs:EmitXfeedTransfer (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h580 Status EmitXfeedTransfer(XfeedKind kind, const Shape& shape,
Dir_emitter.cc432 TF_RETURN_IF_ERROR(EmitXfeedTransfer( in HandleInfeed()
442 EmitXfeedTransfer(XfeedKind::kInfeed, data_shape, data_address)); in HandleInfeed()
448 Status IrEmitter::EmitXfeedTransfer(XfeedKind kind, const Shape& shape, in EmitXfeedTransfer() function in xla::cpu::IrEmitter
527 return EmitXfeedTransfer(XfeedKind::kOutfeed, operand_shape, value); in HandleOutfeed()
538 TF_RETURN_IF_ERROR(EmitXfeedTransfer(XfeedKind::kOutfeed, in HandleOutfeed()