Home
last modified time | relevance | path

Searched defs:infeed (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dxfeed_manager.h113 XfeedQueueManager* infeed() { return &infeed_; } in infeed() function
Dir_emitter.cc326 HloInfeedInstruction* infeed = Cast<HloInfeedInstruction>(instruction); in HandleInfeed() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner.h58 const Literal* infeed = nullptr; member
Dhlo_evaluator.h259 Status HandleInfeed(HloInstruction* infeed) override { in HandleInfeed()
Dhlo_dce_test.cc238 auto infeed = body_builder.AddInstruction( in TEST_F() local
Ddfs_hlo_visitor_with_default.h115 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed()
Dhlo_verifier.cc247 HloInfeedInstruction* infeed = Cast<HloInfeedInstruction>(instruction); in HandleInfeed() local
Dlayout_assignment_test.cc792 auto infeed = false_builder.AddInstruction( in TEST_F() local
Dbuffer_assignment_test.cc2162 auto infeed = in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc138 auto infeed = Infeed(&builder, infeed_shape); in TEST_F() local
227 auto infeed = Infeed(&builder, infeed_shape); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc29 XlaOp infeed = InfeedWithToken(CreateToken(&b), scalar_s32); in BuildComputation() local
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc125 for (const auto& infeed : xfeed_instrs) { in GetXfeedShape() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1473 Status IrEmitterUnnested::HandleInfeed(HloInstruction* infeed) { in HandleInfeed()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1326 XlaOp infeed; in Infeed() local