| /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()
353 inst->fused_instructions_computation()->root_instruction()); in ShouldKeepPrecisionUnchanged()
497 adjust_computation(hlo->fused_instructions_computation(), in AdjustCalledComputationParameters()
555 adjust_computation(hlo->fused_instructions_computation(), hlo); in AdjustCalledComputationRoot()
680 hlo->fused_instructions_computation(), visited_computations); in ResolveInconsistencyOfAliasingBuffersHelper()
746 auto fusion_computation = hlo->fused_instructions_computation(); in ResolveInconsistentFusions()
|
| D | hlo_instructions.cc | 1544 fused_instructions_computation()->unique_id()); in ToProto()
1587 fused_instructions_computation()->parameter_instructions().size()); in AddFusionOperand()
1591 fused_instructions_computation()->AddParameter( in AddFusionOperand()
1610 auto fused_instructions = cloned_fusion->fused_instructions_computation() in MergeFusionInstruction()
1630 instruction_to_merge->fused_instructions_computation() in MergeFusionInstruction()
1660 cloned_fusion->fused_instructions_computation())); in MergeFusionInstruction()
1674 instruction_to_merge->fused_instructions_computation(); in MergeFusionInstructionIntoMultiOutput()
1707 instruction_to_merge->fused_instructions_computation() in MergeFusionInstructionIntoMultiOutput()
1733 HloComputation* HloFusionInstruction::fused_instructions_computation() const { in fused_instructions_computation() function in xla::HloFusionInstruction
1735 auto* fused_instructions_computation = called_computations().front(); in fused_instructions_computation() local
[all …]
|
| D | multi_output_fusion.cc | 420 << instr1->fused_instructions_computation()->ToString( in Perform()
426 << instr2->fused_instructions_computation()->ToString( in Perform()
445 << 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 | 1970 HloComputation* fused_computation = fusion->fused_instructions_computation(); in CheckFusionInstruction()
2024 fusion->fused_instructions_computation()->instructions()) { in CheckFusionInstruction()
2073 {fusion->fused_instructions_computation()})) in CheckFusionInstruction()
2077 << fusion->fused_instructions_computation()->ToString() in CheckFusionInstruction()
2082 TF_RET_CHECK(fused->parent() == fusion->fused_instructions_computation()) in CheckFusionInstruction()
|
| D | hlo_cost_analysis.cc | 846 fusion->fused_instructions_computation()->instructions()) { in HandleFusion()
857 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_computation.cc | 100 CHECK(fusion_instruction_->fused_instructions_computation() == this); in ~HloComputation()
|
| D | hlo.proto | 78 reserved "fused_instructions_computation";
|
| D | hlo_graph_dumper.cc | 563 return ShouldShowSubcomputation(instr->fused_instructions_computation()); in ShouldShowFusionSubcomputation()
|
| D | hlo_instruction.h | 1851 HloComputation* fused_instructions_computation() const;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible.cc | 337 instr.fused_instructions_computation()->MakeInstructionPostOrder()) { in SharedMemoryUsage()
358 instr.fused_instructions_computation()->MakeInstructionPostOrder()) { in NumUnnestedReductions()
504 consumer.fused_instructions_computation()->parameter_instruction( in CreatesNestedLoop()
|
| D | horizontal_loop_fusion.cc | 174 auto fused_instrs = instr.fused_instructions_computation()->instructions(); in HasOnlyRowMajorLayout()
364 ->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/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/tools/ |
| D | hlo_control_flow_flattening.cc | 57 hlo->fused_instructions_computation()->instructions()) { in ExtractInstruction()
|
| 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 | 306 fusion_instruction->fused_instructions_computation()->root_instruction(); in RunDotOutputFusion()
|
| D | dot_op_emitter.cc | 910 hlo.fused_instructions_computation()->root_instruction(); in ProfitableToMakeDotOperandColumnMajor()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 840 ImportAsRegion(*instruction->fused_instructions_computation(), in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 549 *fusion_instr->fused_instructions_computation(), in EmitFusionOp()
|