Home
last modified time | relevance | path

Searched refs:called_computations (Results 1 – 25 of 63) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_graph.h59 absl::Span<HloComputation* const> called_computations, in CallSite() argument
62 called_computations_(called_computations.begin(), in CallSite()
63 called_computations.end()), in CallSite()
70 absl::Span<HloComputation* const> called_computations() const { in called_computations() function
Droot_instruction_sinker_test.cc129 module->entry_computation()->root_instruction()->called_computations()[0]; in TEST_F()
135 ->called_computations()[0] in TEST_F()
161 module->entry_computation()->root_instruction()->called_computations()[0]; in TEST_F()
Dhlo_dce.cc49 for (HloComputation* computation : instruction->called_computations()) { in IsRemovableWhile()
104 for (HloComputation* subcomp : instruction->called_computations()) { in RecursivelyRemoveDeadComputation()
144 for (auto* subcomp : instruction->called_computations()) { in RecursivelyRemoveDeadComputations()
Dcall_graph.cc84 absl::StrJoin(called_computations(), ", ", in ToString()
116 if (!instruction->called_computations().empty()) { in AddCallSiteForInstruction()
121 CallSite(instruction, instruction->called_computations(), context)); in AddCallSiteForInstruction()
124 for (auto* callee : callsites_.back().called_computations()) { in AddCallSiteForInstruction()
217 for (const HloComputation* callee : callsite.called_computations()) { in SetCallContexts()
309 for (auto* callee : callsite.called_computations()) { in Build()
Dconditional_to_select_test.cc73 HloComputation* mapped = root->called_computations()[0]; in TEST_F()
126 HloComputation* mapped = root->called_computations()[0]; in TEST_F()
Dscatter_simplifier.cc135 if (scatter->called_computations().size() != 1) { in ExpandInstruction()
139 scatter->called_computations().size()); in ExpandInstruction()
178 scatter->called_computations().front(), dim_numbers, in ExpandInstruction()
Dcall_inliner.cc130 const auto& callees = call->called_computations(); in Inline()
157 const auto& callees = instruction->called_computations(); in Run()
Dreduce_decomposer.cc60 reduce->called_computations()[0])); in HandleReduce()
103 reduce->called_computations()[0])); in HandleReduce()
Dcall_graph_test.cc235 EXPECT_THAT(call_callsite.called_computations(), in TEST_F()
242 EXPECT_THAT(map_callsite.called_computations(), in TEST_F()
285 EXPECT_THAT(conditional_callsite.called_computations(), in TEST_F()
366 auto called_computations = entry_node.callsites()[0].called_computations(); in TEST_F() local
367 EXPECT_THAT(called_computations, in TEST_F()
Dbfloat16_normalization_test.cc214 EXPECT_EQ(reduce->called_computations().size(), 1); in TEST_F()
215 EXPECT_EQ(reduce->called_computations()[0]->num_parameters(), 2); in TEST_F()
216 EXPECT_EQ(reduce->called_computations()[0] in TEST_F()
221 EXPECT_EQ(reduce->called_computations()[0] in TEST_F()
226 EXPECT_EQ(reduce->called_computations()[0] in TEST_F()
Dhlo_replication_analysis.cc310 auto called = inst->called_computations().front(); in ComputeHloReplicationOnComputation()
320 for (int64_t i = 0; i < inst->called_computations().size(); ++i) { in ComputeHloReplicationOnComputation()
323 inst->called_computations()[i]->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
330 for (auto called : inst->called_computations()) { in ComputeHloReplicationOnComputation()
340 for (auto called : inst->called_computations()) { in ComputeHloReplicationOnComputation()
Dconditional_canonicalizer.cc26 for (auto* branch : conditional->called_computations()) { in CanonicalizeNonTupleConditional()
Dhlo_instructions.cc102 instr->called_computations()) { in SetThreadName()
288 for (HloComputation* computation : called_computations()) { in ClearAsyncComputationInstruction()
297 CHECK(!called_computations().empty()); in async_wrapped_instruction()
298 return called_computations()[0]->root_instruction(); in async_wrapped_instruction()
1573 absl::Span<HloComputation* const> called_computations) in HloCallableInstruction() argument
1579 for (auto called_computation : called_computations) { in HloCallableInstruction()
1587 CHECK(!called_computations().empty()); in called_computation()
1588 return called_computations().front(); in called_computation()
1627 if (called_computations().empty()) { in CloneAndAppendInstructionIntoCalledComputation()
1788 for (auto* comp : called_computations()) { in GetOrCloneCalledComputations()
[all …]
Dhlo_liveness_analysis.cc246 for (auto* caller_comp : caller->called_computations()) { in PropagateLivenessThroughControlFlow()
316 for (auto* called_computation : instruction->called_computations()) { in RunAnalysis()
Dconditional_simplifier.cc588 instruction->called_computations().begin(), in ComputationCallsChannelInstructions()
589 instruction->called_computations().end()); in ComputationCallsChannelInstructions()
598 instruction.called_computations()) { in InstructionCallsChannelInstructions()
Dbfloat16_support.cc119 HloComputation* reduce_comp = hlo.called_computations()[0]; in EffectiveOperandPrecisionIsOutputPrecision()
Dbfloat16_normalization.cc325 for (auto* comp : hlo->called_computations()) { in HandleMultipleOutputs()
396 for (auto* comp : hlo->called_computations()) { in HandleInstruction()
453 if (hlo->called_computations().empty() && in HandleInstruction()
Dhlo_constant_folding.cc58 for (const HloComputation* c : instr->called_computations()) { in IsOrContainsIllegalInstr()
Dflatten_call_graph.cc140 for (auto callee : instruction->called_computations()) { in FlattenNode()
Dhlo_query.cc120 for (const HloComputation* subcomp : instr->called_computations()) { in ContainsInstrWithOpcode()
Dhlo_graph_dumper.cc347 auto it = cluster_ids_.find(instr.called_computations()[0]); in CssIdForInstruction()
712 for (const HloComputation* subcomp : instr->called_computations()) { in DumpComputation()
1436 for (int64_t i = 0; i < instr->called_computations().size(); ++i) { in GetInstructionTrivialComputationStr()
1438 MatchTrivialComputation(instr->called_computations()[i]); in GetInstructionTrivialComputationStr()
1442 if (instr->called_computations().size() == 1) { in GetInstructionTrivialComputationStr()
1504 for (const HloComputation* computation : instr->called_computations()) { in MakeNodeRadiusAroundFilter()
Dhlo_element_type_converter.cc159 TF_RET_CHECK(hlo->called_computations().empty()) << hlo->ToString(); in Run()
Ddynamic_padder.cc79 HloComputation* select = inst->called_computations()[0]; in ChooseIdentityValue()
1211 hlo->shape(), input, init, window, hlo->called_computations()[0])); in RewriteDynamicReduceWindowSamePadding()
1268 input->shape(), input, hlo->called_computations()[0], window, source, in RewriteDynamicSelectAndScatterSamePadding()
1269 init, hlo->called_computations()[1])); in RewriteDynamicSelectAndScatterSamePadding()
1373 sort->called_computations()[0]->num_parameters(); in RewriteDynamicSort()
1383 sort->called_computations()[0]->CloneWithReplacements( in RewriteDynamicSort()
2042 for (auto* sub_computation : hlo->called_computations()) { in DefaultAction()
Dar_crs_combiner.cc169 computation_is_addition(instruction->called_computations()[0]) && in MatchesArCrsPattern()
182 computation_is_addition(next->called_computations()[0])) { in MatchesArCrsPattern()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc310 *instr->called_computations()[0], &reduce_op.body(), &builder_, in CreateOpInFusion()
481 *sort_instr->called_computations()[0], &sort.getComparator(), &builder_)); in EmitSortOp()
651 *scatter_instr->called_computations()[0], &scatter.getUpdateComputation(), in EmitScatterOp()
1181 *instr->called_computations()[0], &all_reduce_op.getComputation(), in EmitAllReduceOp()
1206 *instr->called_computations()[0], &all_reduce_start_op.getComputation(), in EmitAllReduceStartOp()
1242 *instr->called_computations()[0], &reduce_scatter_op.getComputation(), in EmitReduceScatterOp()
1391 TF_RETURN_IF_ERROR(ImportAsLmhloRegion(instr->called_computations()[i], in EmitCaseOp()
1403 instr->called_computations()[1]->root_instruction(), &operands)); in EmitWhileOp()
1417 TF_RETURN_IF_ERROR(ImportAsLmhloRegion(instr->called_computations()[1], in EmitWhileOp()
1420 TF_RETURN_IF_ERROR(ImportAsLmhloRegion(instr->called_computations()[0], in EmitWhileOp()

123