| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | xfeed_manager.h | 112 XfeedQueueManager* infeed() { return &infeed_; } in infeed() function
|
| D | ir_emitter.cc | 403 HloInfeedInstruction* infeed = Cast<HloInfeedInstruction>(instruction); in HandleInfeed() 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 | 93 auto infeed = InfeedWithToken(token, ShapeUtil::MakeShape(F32, {4, 8}), ""); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 159 for (const auto& infeed : xfeed_instrs) { in GetXfeedShape() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dce_test.cc | 274 auto infeed = body_builder.AddInstruction( in TEST_F() local
|
| D | dfs_hlo_visitor_with_default.h | 147 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed()
|
| D | hlo_cost_analysis.cc | 339 Status HloCostAnalysis::HandleInfeed(const HloInstruction* infeed) { in HandleInfeed()
|
| D | hlo_cost_analysis_test.cc | 950 HloInstruction* infeed = in TEST_F() local
|
| D | layout_assignment_test.cc | 752 auto infeed = false_builder.AddInstruction( in TEST_F() local
|
| D | hlo_verifier.cc | 772 HloInfeedInstruction* infeed = Cast<HloInfeedInstruction>(instruction); in HandleInfeed() local
|
| D | buffer_assignment_test.cc | 2388 auto infeed = in TEST_F() local
|
| D | hlo_evaluator.cc | 1115 Status HloEvaluator::HandleInfeed(HloInstruction* infeed) { in HandleInfeed()
|
| D | hlo_evaluator_test.cc | 4877 HloInstruction* infeed = b.AddInstruction(HloInstruction::CreateInfeed( in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1265 const HloInfeedInstruction* infeed = xla::Cast<HloInfeedInstruction>(instr); in EmitInfeedOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3326 auto infeed = branch_b.AddInstruction(HloInstruction::CreateInfeed( in HandleInfeed() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1792 XlaOp infeed; in Infeed() local
|