/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A 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() 89 inner_fusion->fused_instructions_computation(), in TEST_F() 90 outer_fusion->fused_instructions_computation()}) { in TEST_F() 101 inner_fusion->fused_instructions_computation()->instructions()) { in TEST_F()
|
H A 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() 121 fusion->fused_instructions_computation()->MakeInstructionPostOrder()) { in RevertIfFusionInternalBF16Changes() 349 inst->fused_instructions_computation()->root_instruction()); in ShouldKeepPrecisionUnchanged() 492 adjust_computation(hlo->fused_instructions_computation(), in AdjustCalledComputationParameters() 550 adjust_computation(hlo->fused_instructions_computation(), hlo); in AdjustCalledComputationRoot() 676 hlo->fused_instructions_computation(), visited_computations); in ResolveInconsistencyOfAliasingBuffersHelper() 745 auto fusion_computation = hlo->fused_instructions_computation(); in ResolveInconsistentFusions()
|
H A D | hlo_instructions.cc | 1874 fused_instructions_computation()->unique_id()); in ToProto() 1931 auto fused_instructions = cloned_fusion->fused_instructions_computation() in MergeFusionInstruction() 1951 instruction_to_merge->fused_instructions_computation() in MergeFusionInstruction() 1981 cloned_fusion->fused_instructions_computation())); in MergeFusionInstruction() 1995 instruction_to_merge->fused_instructions_computation(); in MergeFusionInstructionIntoMultiOutput() 2028 instruction_to_merge->fused_instructions_computation() in MergeFusionInstructionIntoMultiOutput() 2054 HloComputation* HloFusionInstruction::fused_instructions_computation() const { in fused_instructions_computation() function in xla::HloFusionInstruction 2056 auto* fused_instructions_computation = called_computations().front(); in fused_instructions_computation() local 2057 CHECK(fused_instructions_computation->IsFusionComputation()) in fused_instructions_computation() 2058 << "Computation " << fused_instructions_computation->name() in fused_instructions_computation() [all …]
|
H A 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()
|
H A D | defuser.cc | 48 fusion_instruction->fused_instructions_computation(); in Defuse()
|
H A D | fusion_node_indexing_evaluation.cc | 131 fusion_->fused_instructions_computation()->MakeInstructionPostOrder(); in RecomputeCache()
|
H A D | bitcast_decomposer_test.cc | 171 ->fused_instructions_computation() in TEST_P()
|
H A D | hlo_verifier.cc | 2190 HloComputation* fused_computation = fusion->fused_instructions_computation(); in CheckFusionInstruction() 2244 fusion->fused_instructions_computation()->instructions()) { in CheckFusionInstruction() 2293 {fusion->fused_instructions_computation()})) in CheckFusionInstruction() 2297 << fusion->fused_instructions_computation()->ToString() in CheckFusionInstruction() 2302 TF_RET_CHECK(fused->parent() == fusion->fused_instructions_computation()) in CheckFusionInstruction()
|
H A D | defuser_test.cc | 207 outer_fusion->fused_instructions_computation()->CreateFusionInstruction( in TEST_F()
|
H A D | hlo_cost_analysis.cc | 872 fusion->fused_instructions_computation()->instructions()) { in HandleFusion() 883 ProcessSubcomputation(fusion->fused_instructions_computation())); in HandleFusion()
|
H A D | hlo_cse_test.cc | 489 ->fused_instructions_computation(); in TEST_F()
|
H A D | instruction_fusion.cc | 834 hlo->fused_instructions_computation()->instructions()) { in ExtractInstruction()
|
H A D | hlo_instruction_test.cc | 937 auto* fused_computation = fusion->fused_instructions_computation(); in TEST_F() 1422 EXPECT_EQ(fusion->fused_instructions_computation()->num_parameters(), 1); in TEST_F()
|
H A D | layout_assignment.cc | 1914 fusion->fused_instructions_computation()->MakeInstructionPostOrder()) { in SetFusionLayouts() 2142 instruction->fused_instructions_computation()), in CalculateComputationLayout()
|
H A D | hlo_computation.cc | 108 CHECK(fusion_instruction_->fused_instructions_computation() == this); in ~HloComputation()
|
H A D | hlo.proto | 98 reserved "fused_instructions_computation";
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | gpu_fusible.cc | 316 instr.fused_instructions_computation()->instructions()) { in SharedMemoryUsageNoCache() 358 instr.fused_instructions_computation()->instructions()) { in NumUnnestedReductionsNoCache() 528 consumer.fused_instructions_computation()->parameter_instruction( in CreatesNestedLoop()
|
H A D | fusion_bitcast_lift.cc | 86 fusion->fused_instructions_computation())) { in Run() 253 cloned_fusion->fused_instructions_computation()->Accept( in Run()
|
H A D | horizontal_loop_fusion.cc | 179 auto fused_instrs = instr.fused_instructions_computation()->instructions(); in HasOnlyRowMajorLayout() 370 ->fused_instructions_computation() in CreateFusedComputation()
|
H A D | hlo_fusion_stats.cc | 122 GetUniqueOpcodes(fusion->fused_instructions_computation()); in HandleFusion()
|
H A D | instruction_fusion_test.cc | 561 EXPECT_THAT(root->fused_instructions_computation()->root_instruction(), in TEST_F()
|
H A D | ir_emission_utils.cc | 775 instr.fused_instructions_computation(), input_ids) in ShmemTransposeSupportedForInputs()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
H A D | cpu_fusion_test.cc | 205 << fusion_instruction1->fused_instructions_computation()->ToString(); in TEST_F() 214 << fusion_instruction2->fused_instructions_computation()->ToString(); in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
H A D | interactive_graphviz.cc | 355 << instr->fused_instructions_computation()->name() << std::endl; in DoInfoCommand()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | cpu_layout_assignment_test.cc | 306 fusion_instruction->fused_instructions_computation()->root_instruction(); in RunDotOutputFusion()
|