| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_graph.cc | 79 absl::StrJoin(called_computations(), ", ", in ToString()
109 if (!instruction->called_computations().empty()) { in AddCallSiteForInstruction()
114 CallSite(instruction, instruction->called_computations(), context)); in AddCallSiteForInstruction()
117 for (auto* callee : callsites_.back().called_computations()) { in AddCallSiteForInstruction()
210 for (const HloComputation* callee : callsite.called_computations()) { in SetCallContexts()
302 for (auto* callee : callsite.called_computations()) { in Build()
|
| D | call_graph.h | 62 const std::vector<HloComputation*>& called_computations, in CallSite() argument
65 called_computations_(called_computations), in CallSite()
72 const std::vector<HloComputation*>& called_computations() const { in called_computations() function
|
| D | bfloat16_normalization_test.cc | 212 EXPECT_EQ(reduce->called_computations().size(), 1); in TEST_F()
213 EXPECT_EQ(reduce->called_computations()[0]->num_parameters(), 2); in TEST_F()
214 EXPECT_EQ(reduce->called_computations()[0] in TEST_F()
219 EXPECT_EQ(reduce->called_computations()[0] in TEST_F()
224 EXPECT_EQ(reduce->called_computations()[0] in TEST_F()
|
| D | call_graph_test.cc | 230 EXPECT_THAT(call_callsite.called_computations(), in TEST_F()
237 EXPECT_THAT(map_callsite.called_computations(), in TEST_F()
280 EXPECT_THAT(conditional_callsite.called_computations(), in TEST_F()
361 const std::vector<HloComputation*>& called_computations = in TEST_F() local
362 entry_node.callsites()[0].called_computations(); in TEST_F()
363 EXPECT_THAT(called_computations, in TEST_F()
|
| D | bfloat16_normalization.cc | 214 for (auto* comp : hlo->called_computations()) { in HandleMultipleOutputs()
288 for (auto* comp : hlo->called_computations()) { in HandleInstruction()
336 if (hlo->called_computations().empty() && in HandleInstruction()
|
| D | hlo_dce.cc | 75 for (auto* subcomp : instruction->called_computations()) { in Run()
|
| D | hlo_constant_folding.cc | 58 for (const HloComputation* c : instr->called_computations()) { in IsOrContainsIllegalInstr()
|
| D | hlo_query.cc | 113 for (const HloComputation* subcomp : instr->called_computations()) { in ContainsInstrWithOpcode()
|
| D | bfloat16_support.cc | 113 HloComputation* reduce_comp = hlo.called_computations()[0]; in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | flatten_call_graph.cc | 103 for (auto callee : instruction->called_computations()) { in FlattenNode()
|
| D | ar_crs_combiner.cc | 73 !computation_is_addition(instruction->called_computations()[0]) || in MatchesArCrsPattern()
88 computation_is_addition(next->called_computations()[0])) { in MatchesArCrsPattern()
|
| D | call_inliner.cc | 129 const auto& callees = call->called_computations(); in Inline()
|
| D | hlo_graph_dumper.cc | 670 for (const HloComputation* subcomp : instr->called_computations()) { in DumpComputation()
1235 for (int64 i = 0; i < instr->called_computations().size(); ++i) { in GetInstructionTrivialComputationStr()
1237 MatchTrivialComputation(instr->called_computations()[i]); in GetInstructionTrivialComputationStr()
1241 if (instr->called_computations().size() == 1) { in GetInstructionTrivialComputationStr()
1299 for (const HloComputation* computation : instr->called_computations()) { in MakeNodeRadiusAroundFilter()
|
| D | hlo_element_type_converter.cc | 152 TF_RET_CHECK(hlo->called_computations().empty()) << hlo->ToString(); in Run()
|
| D | hlo_liveness_analysis.cc | 297 for (auto* called_computation : instruction->called_computations()) { in RunAnalysis()
|
| D | hlo_verifier.cc | 86 if (!hlo->called_computations().empty() && !IsCallerInstruction(hlo)) { in Preprocess()
1282 TF_RET_CHECK(fusion->called_computations() == in CheckFusionInstruction()
1290 << ComputationsToString(fusion->called_computations()); in CheckFusionInstruction()
|
| D | heap_simulator.cc | 300 instruction->called_computations()) { in RunComputation()
535 for (const auto* c : instruction->called_computations()) { in AccountForSubcomputationMemory()
|
| D | hlo_rematerialization.cc | 964 for (const HloComputation* computation : callsite->called_computations()) { in CalledComputationsMemoryUsage()
1146 callsite->called_computations()) { in RematerializeComputation()
|
| D | hlo_memory_scheduler.cc | 233 for (const auto* c : instruction->called_computations()) { in BytesFreedIfScheduled()
|
| D | hlo_instruction_test.cc | 779 EXPECT_THAT(fusion->called_computations(), ElementsAre(fused_computation)); in TEST_F()
782 EXPECT_THAT(fusion->called_computations(), ElementsAre(fused_computation)); in TEST_F()
785 EXPECT_THAT(fusion->called_computations(), ElementsAre(fused_computation)); in TEST_F()
|
| D | hlo_instruction.cc | 1260 for (const auto& computation : called_computations()) { in HasSideEffect()
2314 } else if (!called_computations().empty()) { in ExtraAttributesToString()
2317 StrJoin(called_computations(), ", ", in ExtraAttributesToString()
2363 if (!called_computations().empty()) { in ExtraAttributesToString()
2366 StrJoin(called_computations(), ", ", in ExtraAttributesToString()
|
| D | hlo_module.cc | 507 instruction->called_computations()) { in MakeComputationPostOrder()
|
| D | hlo_instructions.h | 1130 return called_computations()[kSelectComputationIndex]; in select()
1134 return called_computations()[kScatterComputationIndex]; in scatter()
|
| D | hlo_instructions.cc | 1328 CHECK(!called_computations().empty()); in fused_instructions_computation()
1329 auto* fused_instructions_computation = called_computations().front(); in fused_instructions_computation()
1385 if (called_computations().empty()) { in CloneAndFuseInternal()
|
| D | hlo_dataflow_analysis.cc | 689 for (HloComputation* called_computation : user->called_computations()) { in Propagate()
|