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.h554 Status EmitXfeedTransfer(XfeedKind kind, const Shape& shape,
Dir_emitter.cc368 TF_RETURN_IF_ERROR(EmitXfeedTransfer( in HandleInfeed()
378 EmitXfeedTransfer(XfeedKind::kInfeed, data_shape, data_address)); in HandleInfeed()
384 Status IrEmitter::EmitXfeedTransfer(XfeedKind kind, const Shape& shape, in EmitXfeedTransfer() function in xla::cpu::IrEmitter
486 return EmitXfeedTransfer(XfeedKind::kOutfeed, operand_shape, value); in HandleOutfeed()
497 TF_RETURN_IF_ERROR(EmitXfeedTransfer(XfeedKind::kOutfeed, in HandleOutfeed()