| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_graph_dumper_test.cc | 70 for (auto* instr : outer_fusion->fused_instructions_computation() in TEST_F()
79 outer_fusion->fused_instructions_computation()->CreateFusionInstruction( in TEST_F()
88 inner_fusion->fused_instructions_computation(), in TEST_F()
89 outer_fusion->fused_instructions_computation()}) { in TEST_F()
100 inner_fusion->fused_instructions_computation()->instructions()) { in TEST_F()
|
| D | bfloat16_propagation.cc | 51 auto root = fusion->fused_instructions_computation()->root_instruction(); in DetermineFusionComputationPrecision()
69 fusion->fused_instructions_computation()->MakeInstructionPostOrder(); in DetermineFusionComputationPrecision()
74 fusion->fused_instructions_computation()); in DetermineFusionComputationPrecision()
86 auto root = fusion->fused_instructions_computation()->root_instruction(); in RevertIfFusionInternalBF16Changes()
125 fusion->fused_instructions_computation()->MakeInstructionPostOrder()) { in RevertIfFusionInternalBF16Changes()
356 inst->fused_instructions_computation()->root_instruction()); in ShouldKeepPrecisionUnchanged()
502 adjust_computation(hlo->fused_instructions_computation(), in AdjustCalledComputationParameters()
560 adjust_computation(hlo->fused_instructions_computation(), hlo); in AdjustCalledComputationRoot()
685 hlo->fused_instructions_computation(), visited_computations); in ResolveInconsistencyOfAliasingBuffersHelper()
751 auto fusion_computation = hlo->fused_instructions_computation(); in ResolveInconsistentFusions()
|
| D | hlo_instructions.cc | 1407 fused_instructions_computation()->unique_id()); in ToProto()
1450 fused_instructions_computation()->parameter_instructions().size()); in AddFusionOperand()
1454 fused_instructions_computation()->AddParameter( in AddFusionOperand()
1473 auto fused_instructions = cloned_fusion->fused_instructions_computation() in MergeFusionInstruction()
1493 instruction_to_merge->fused_instructions_computation() in MergeFusionInstruction()
1523 cloned_fusion->fused_instructions_computation())); in MergeFusionInstruction()
1537 instruction_to_merge->fused_instructions_computation(); in MergeFusionInstructionIntoMultiOutput()
1570 instruction_to_merge->fused_instructions_computation() in MergeFusionInstructionIntoMultiOutput()
1596 HloComputation* HloFusionInstruction::fused_instructions_computation() const { in fused_instructions_computation() function in xla::HloFusionInstruction
1598 auto* fused_instructions_computation = called_computations().front(); in fused_instructions_computation() local
[all …]
|
| D | multi_output_fusion.cc | 422 << instr1->fused_instructions_computation()->ToString( in Perform()
428 << instr2->fused_instructions_computation()->ToString( in Perform()
447 << fusion->fused_instructions_computation()->ToString( in Perform()
|
| D | defuser.cc | 49 fusion_instruction->fused_instructions_computation(); in Defuse()
|
| D | fusion_node_indexing_evaluation.cc | 131 fusion_->fused_instructions_computation()->MakeInstructionPostOrder(); in RecomputeCache()
|
| D | defuser_test.cc | 207 outer_fusion->fused_instructions_computation()->CreateFusionInstruction( in TEST_F()
|
| D | hlo_verifier.cc | 1618 HloComputation* fused_computation = fusion->fused_instructions_computation(); in CheckFusionInstruction()
1672 fusion->fused_instructions_computation()->instructions()) { in CheckFusionInstruction()
1721 {fusion->fused_instructions_computation()})) in CheckFusionInstruction()
1725 << fusion->fused_instructions_computation()->ToString() in CheckFusionInstruction()
1730 TF_RET_CHECK(fused->parent() == fusion->fused_instructions_computation()) in CheckFusionInstruction()
|
| D | hlo_cost_analysis.cc | 817 fusion->fused_instructions_computation()->instructions()) { in HandleFusion()
828 ProcessSubcomputation(fusion->fused_instructions_computation())); in HandleFusion()
|
| D | hlo_cse_test.cc | 511 ->fused_instructions_computation(); in TEST_F()
|
| D | hlo_instruction_test.cc | 838 auto* fused_computation = fusion->fused_instructions_computation(); in TEST_F()
1323 EXPECT_EQ(fusion->fused_instructions_computation()->num_parameters(), 1); in TEST_F()
|
| D | hlo.proto | 43 reserved "fused_instructions_computation";
|
| D | hlo_graph_dumper.cc | 563 return ShouldShowSubcomputation(instr->fused_instructions_computation()); in ShouldShowFusionSubcomputation()
|
| D | hlo_instruction.h | 1732 HloComputation* fused_instructions_computation() const;
|
| D | hlo_instructions.h | 941 HloComputation* fused_instructions_computation() const;
|
| D | hlo_instruction.cc | 4013 HloComputation* HloInstruction::fused_instructions_computation() const { in fused_instructions_computation() function in xla::HloInstruction
4014 return Cast<HloFusionInstruction>(this)->fused_instructions_computation(); in fused_instructions_computation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_fusion_test.cc | 206 << fusion_instruction1->fused_instructions_computation()->ToString(); in TEST_F()
215 << fusion_instruction2->fused_instructions_computation()->ToString(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible.cc | 338 instr.fused_instructions_computation()->MakeInstructionPostOrder()) { in SharedMemoryUsage()
478 consumer.fused_instructions_computation()->parameter_instruction( in CreatesNestedLoop()
|
| D | horizontal_loop_fusion.cc | 194 auto instrs = fusion_instr.fused_instructions_computation()->instructions(); in HasOnlyRowMajorLayout()
359 ->fused_instructions_computation() in CreateFusedComputation()
|
| D | instruction_fusion_test.cc | 561 EXPECT_THAT(root->fused_instructions_computation()->root_instruction(), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | interactive_graphviz.cc | 345 << instr->fused_instructions_computation()->name() << std::endl; in DoInfoCommand()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 307 fusion_instruction->fused_instructions_computation()->root_instruction(); in RunDotOutputFusion()
|
| D | dot_op_emitter.cc | 905 hlo.fused_instructions_computation()->root_instruction(); in ProfitableToMakeDotOperandColumnMajor()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 805 ImportAsRegion(*instruction->fused_instructions_computation(), in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 498 *fusion_instr->fused_instructions_computation(), in EmitFusionOp()
|