Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dconditional_to_select_test.cc74 HloComputation* mapped = root->called_computations()[0]; in TEST_F()
127 HloComputation* mapped = root->called_computations()[0]; in TEST_F()
183 HloComputation* mapped = root->called_computations()[0]; in TEST_F()
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()
Dcall_graph.cc80 absl::StrJoin(called_computations(), ", ", in ToString()
112 if (!instruction->called_computations().empty()) { in AddCallSiteForInstruction()
117 CallSite(instruction, instruction->called_computations(), context)); in AddCallSiteForInstruction()
120 for (auto* callee : callsites_.back().called_computations()) { in AddCallSiteForInstruction()
213 for (const HloComputation* callee : callsite.called_computations()) { in SetCallContexts()
305 for (auto* callee : callsite.called_computations()) { in Build()
Dcall_graph.h62 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
Dhlo_replication_analysis.cc244 auto called = inst->called_computations().front(); in ComputeHloReplicationOnComputation()
254 for (int64 i = 0; i < inst->called_computations().size(); ++i) { in ComputeHloReplicationOnComputation()
257 inst->called_computations()[i]->parameter_instruction(0)); in ComputeHloReplicationOnComputation()
262 for (auto called : inst->called_computations()) { in ComputeHloReplicationOnComputation()
270 for (auto called : inst->called_computations()) { in ComputeHloReplicationOnComputation()
Dcall_graph_test.cc230 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()
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()
Dconditional_canonicalizer.cc26 for (auto* branch : conditional->called_computations()) { in CanonicalizeNonTupleConditional()
Dhlo_dce.cc98 for (auto* subcomp : instruction->called_computations()) { in Run()
Dhlo_constant_folding.cc58 for (const HloComputation* c : instr->called_computations()) { in IsOrContainsIllegalInstr()
Dbfloat16_support.cc116 HloComputation* reduce_comp = hlo.called_computations()[0]; in EffectiveOperandPrecisionIsOutputPrecision()
Dflatten_call_graph.cc103 for (auto callee : instruction->called_computations()) { in FlattenNode()
Dbfloat16_normalization.cc326 for (auto* comp : hlo->called_computations()) { in HandleMultipleOutputs()
397 for (auto* comp : hlo->called_computations()) { in HandleInstruction()
454 if (hlo->called_computations().empty() && in HandleInstruction()
Dcall_inliner.cc127 const auto& callees = call->called_computations(); in Inline()
Dhlo_query.cc115 for (const HloComputation* subcomp : instr->called_computations()) { in ContainsInstrWithOpcode()
Dhlo_cse.cc114 for (auto c : instruction->called_computations()) { in CseHash()
Ddynamic_padder.cc72 HloComputation* select = inst->called_computations()[0]; in ChooseIdentityValue()
1056 hlo->called_computations()[0]), in RewriteDynamicReduceWindowSamePadding()
1115 input->shape(), input, hlo->called_computations()[0], window, source, in RewriteDynamicSelectAndScatterSamePadding()
1116 init, hlo->called_computations()[1]), in RewriteDynamicSelectAndScatterSamePadding()
1224 sort->called_computations()[0]->num_parameters(); in RewriteDynamicSort()
1234 sort->called_computations()[0]->CloneWithReplacements( in RewriteDynamicSort()
1696 for (auto* sub_computation : hlo->called_computations()) { in DefaultAction()
Dhlo_graph_dumper.cc677 for (const HloComputation* subcomp : instr->called_computations()) { in DumpComputation()
1274 for (int64 i = 0; i < instr->called_computations().size(); ++i) { in GetInstructionTrivialComputationStr()
1276 MatchTrivialComputation(instr->called_computations()[i]); in GetInstructionTrivialComputationStr()
1280 if (instr->called_computations().size() == 1) { in GetInstructionTrivialComputationStr()
1338 for (const HloComputation* computation : instr->called_computations()) { in MakeNodeRadiusAroundFilter()
Dhlo_element_type_converter.cc152 TF_RET_CHECK(hlo->called_computations().empty()) << hlo->ToString(); in Run()
Dhlo_live_range.cc107 instruction->called_computations()) { in FlattenSchedule()
Dar_crs_combiner.cc170 computation_is_addition(instruction->called_computations()[0]) && in MatchesArCrsPattern()
183 computation_is_addition(next->called_computations()[0])) { in MatchesArCrsPattern()
Dhlo_instructions.cc1597 CHECK(!called_computations().empty()); in fused_instructions_computation()
1598 auto* fused_instructions_computation = called_computations().front(); in fused_instructions_computation()
1654 if (called_computations().empty()) { in CloneAndFuseInternal()
2374 absl::Span<HloComputation* const> called_computations, in HloCustomCallInstruction() argument
2387 for (auto comp : called_computations) { in HloCustomCallInstruction()
2559 if (called_computations().size() != other.called_computations().size()) { in IdenticalSlowPath()
2562 for (int64 i = 0; i < called_computations().size(); ++i) { in IdenticalSlowPath()
2563 if (!eq_computations(called_computations()[i], in IdenticalSlowPath()
2564 other.called_computations()[i])) { in IdenticalSlowPath()
Dhlo_liveness_analysis.cc297 for (auto* called_computation : instruction->called_computations()) { in RunAnalysis()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc423 *sort_instr->called_computations()[0], &sort.comparator(), &builder_)); in EmitSortOp()
576 *scatter_instr->called_computations()[0], &scatter.update_computation(), in EmitScatterOp()
954 *instr->called_computations()[0], &reduce_op.body(), &builder_)); in EmitReduceOp()
966 *instr->called_computations()[0], &map_op.computation(), &builder_)); in EmitMapOp()
1085 *instr->called_computations()[0], &all_reduce_op.computation(), in EmitAllReduceOp()
1225 *hlo_reduce_window->called_computations()[0], &reduce_window.body(), in EmitReduceWindowOp()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc241 fusion->called_computations()[0], fusion_output_array, fused_emitter, in EmitFusedDynamicUpdateSliceInPlace()

12