Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h196 Status HandleInfeed(HloInstruction* xla_infeed) override;
Dir_emitter_unnested.cc3190 Status IrEmitterUnnested::HandleInfeed(HloInstruction* xla_infeed) { in HandleInfeed() argument
3191 TF_ASSIGN_OR_RETURN(auto input, GetMlirEmitterInput(xla_infeed)); in HandleInfeed()