Home
last modified time | relevance | path

Searched refs:fused_instructions_computation (Results 1 – 25 of 35) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_graph_dumper_test.cc70 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 Dbfloat16_propagation.cc51 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 Dhlo_instructions.cc1874 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 Dmulti_output_fusion.cc422 << 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 Ddefuser.cc48 fusion_instruction->fused_instructions_computation(); in Defuse()
H A Dfusion_node_indexing_evaluation.cc131 fusion_->fused_instructions_computation()->MakeInstructionPostOrder(); in RecomputeCache()
H A Dbitcast_decomposer_test.cc171 ->fused_instructions_computation() in TEST_P()
H A Dhlo_verifier.cc2190 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 Ddefuser_test.cc207 outer_fusion->fused_instructions_computation()->CreateFusionInstruction( in TEST_F()
H A Dhlo_cost_analysis.cc872 fusion->fused_instructions_computation()->instructions()) { in HandleFusion()
883 ProcessSubcomputation(fusion->fused_instructions_computation())); in HandleFusion()
H A Dhlo_cse_test.cc489 ->fused_instructions_computation(); in TEST_F()
H A Dinstruction_fusion.cc834 hlo->fused_instructions_computation()->instructions()) { in ExtractInstruction()
H A Dhlo_instruction_test.cc937 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 Dlayout_assignment.cc1914 fusion->fused_instructions_computation()->MakeInstructionPostOrder()) { in SetFusionLayouts()
2142 instruction->fused_instructions_computation()), in CalculateComputationLayout()
H A Dhlo_computation.cc108 CHECK(fusion_instruction_->fused_instructions_computation() == this); in ~HloComputation()
H A Dhlo.proto98 reserved "fused_instructions_computation";
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_fusible.cc316 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 Dfusion_bitcast_lift.cc86 fusion->fused_instructions_computation())) { in Run()
253 cloned_fusion->fused_instructions_computation()->Accept( in Run()
H A Dhorizontal_loop_fusion.cc179 auto fused_instrs = instr.fused_instructions_computation()->instructions(); in HasOnlyRowMajorLayout()
370 ->fused_instructions_computation() in CreateFusedComputation()
H A Dhlo_fusion_stats.cc122 GetUniqueOpcodes(fusion->fused_instructions_computation()); in HandleFusion()
H A Dinstruction_fusion_test.cc561 EXPECT_THAT(root->fused_instructions_computation()->root_instruction(), in TEST_F()
H A Dir_emission_utils.cc775 instr.fused_instructions_computation(), input_ids) in ShmemTransposeSupportedForInputs()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
H A Dcpu_fusion_test.cc205 << 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 Dinteractive_graphviz.cc355 << instr->fused_instructions_computation()->name() << std::endl; in DoInfoCommand()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_layout_assignment_test.cc306 fusion_instruction->fused_instructions_computation()->root_instruction(); in RunDotOutputFusion()

12