| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | xfeed_manager.h | 113 XfeedQueueManager* infeed() { return &infeed_; } in infeed() function
|
| D | ir_emitter.cc | 390 HloInfeedInstruction* infeed = Cast<HloInfeedInstruction>(instruction); in HandleInfeed() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner_interface.h | 55 const Literal* infeed = nullptr; member
|
| D | hlo_evaluator.h | 283 Status HandleInfeed(HloInstruction* infeed) override { in HandleInfeed()
|
| D | hlo_dce_test.cc | 276 auto infeed = body_builder.AddInstruction( in TEST_F() local
|
| D | dfs_hlo_visitor_with_default.h | 134 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed()
|
| D | hlo_cost_analysis_test.cc | 814 HloInstruction* infeed = in TEST_F() local
|
| D | hlo_cost_analysis.cc | 339 Status HloCostAnalysis::HandleInfeed(const HloInstruction* infeed) { in HandleInfeed()
|
| D | layout_assignment_test.cc | 789 auto infeed = false_builder.AddInstruction( in TEST_F() local
|
| D | hlo_verifier.cc | 422 HloInfeedInstruction* infeed = Cast<HloInfeedInstruction>(instruction); in HandleInfeed() local
|
| D | buffer_assignment_test.cc | 2310 auto infeed = in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 138 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/ |
| D | multiple_devices_on_host_test.cc | 29 XlaOp infeed = InfeedWithToken(CreateToken(&b), scalar_s32); in BuildComputation() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 83 auto infeed = InfeedWithToken(token, ShapeUtil::MakeShape(F32, {4, 8}), ""); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 150 for (const auto& infeed : xfeed_instrs) { in GetXfeedShape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1106 const HloInfeedInstruction* infeed = xla::Cast<HloInfeedInstruction>(instr); in EmitInfeedOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2570 auto infeed = branch_b.AddInstruction(HloInstruction::CreateInfeed( in HandleInfeed() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1707 XlaOp infeed; in Infeed() local
|