| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_graph.h | 62 const std::vector<HloComputation*>& called_computations, in CallSite()
72 const std::vector<HloComputation*>& called_computations() const { in called_computations()
86 const std::vector<HloComputation*> called_computations_;
95 CallGraphNode(HloComputation* computation);
98 HloComputation* computation() const { return computation_; } in computation()
111 const std::vector<HloComputation*>& callees() const { return callees_; } in callees()
119 const std::vector<HloComputation*>& callers() const { return callers_; } in callers()
143 HloComputation* computation_;
147 std::vector<HloComputation*> callees_;
148 tensorflow::gtl::FlatSet<HloComputation*> callee_set_;
[all …]
|
| D | hlo_module.h | 63 HloComputation* AddEntryComputation(
64 std::unique_ptr<HloComputation> computation);
67 HloComputation* AddEmbeddedComputation(
68 std::unique_ptr<HloComputation> computation);
71 Status RemoveEmbeddedComputation(HloComputation* to_remove);
81 const std::unordered_map<HloComputation*, HloComputation*>& replacements);
90 HloComputation* DeepCloneComputation(HloComputation* computation);
93 const HloComputation* entry_computation() const { in entry_computation()
97 HloComputation* entry_computation() { in entry_computation()
122 std::vector<std::unique_ptr<HloComputation>>::const_iterator>>
[all …]
|
| D | while_loop_simplifier_test.cc | 31 HloComputation* MakeSimpleLoop(int num_iters, HloModule* module);
35 HloComputation* MakeAlwaysTrueComputation(const Shape& param_shape,
39 HloComputation* WhileLoopSimplifierTest::MakeSimpleLoop(int num_iters, in MakeSimpleLoop()
41 HloComputation::Builder builder(TestName()); in MakeSimpleLoop()
50 HloComputation* condition; in MakeSimpleLoop()
52 HloComputation::Builder cond_builder(TestName() + ".condition"); in MakeSimpleLoop()
66 HloComputation* body; in MakeSimpleLoop()
68 HloComputation::Builder body_builder(TestName() + ".body"); in MakeSimpleLoop()
97 HloComputation* WhileLoopSimplifierTest::MakeAlwaysTrueComputation( in MakeAlwaysTrueComputation()
99 HloComputation::Builder builder(TestName() + ".always_true"); in MakeAlwaysTrueComputation()
[all …]
|
| D | hlo_computation.cc | 47 std::unique_ptr<HloComputation> HloComputation::Builder::Build( in Build()
59 return WrapUnique(new HloComputation(name_, parameter_count, &instructions_, in Build()
63 HloComputation::HloComputation( in HloComputation() function in xla::HloComputation
90 HloInstruction* HloComputation::AddInstruction( in AddInstruction()
98 HloInstruction* HloComputation::AddInstructionInternal( in AddInstructionInternal()
111 HloInstruction* HloComputation::AddParameter( in AddParameter()
122 Status HloComputation::RemoveParameter(int64 param_no) { in RemoveParameter()
161 void HloComputation::Reparent(HloInstruction* instruction) { in Reparent()
165 bool HloComputation::IsRemovable(const HloInstruction* instruction) { in IsRemovable()
182 bool HloComputation::HasSideEffect() const { in HasSideEffect()
[all …]
|
| D | hlo_module.cc | 52 HloComputation* HloModule::AddComputationInternal( in AddComputationInternal()
53 std::unique_ptr<HloComputation> computation, bool is_entry, in AddComputationInternal()
91 HloComputation* HloModule::AddEntryComputation( in AddEntryComputation()
92 std::unique_ptr<HloComputation> computation) { in AddEntryComputation()
97 Status HloModule::RemoveEmbeddedComputation(HloComputation* to_remove) { in RemoveEmbeddedComputation()
100 [&to_remove](const std::unique_ptr<HloComputation>& comp) { in RemoveEmbeddedComputation()
108 HloComputation* HloModule::AddEmbeddedComputation( in AddEmbeddedComputation()
109 std::unique_ptr<HloComputation> computation) { in AddEmbeddedComputation()
115 const std::unordered_map<HloComputation*, HloComputation*>& replacements) { in ReplaceComputations() argument
118 std::vector<std::unique_ptr<HloComputation>> new_computations; in ReplaceComputations()
[all …]
|
| D | flatten_call_graph_test.cc | 36 std::unique_ptr<HloComputation> MakeScalarComputation() { in MakeScalarComputation()
37 HloComputation::Builder builder(TestName() + ".ScalarComputation"); in MakeScalarComputation()
47 std::unique_ptr<HloComputation> MakeMappingComputation( in MakeMappingComputation()
48 HloComputation* map_computation, int64 callsites) { in MakeMappingComputation()
49 HloComputation::Builder builder(TestName() + ".MappingComputation"); in MakeMappingComputation()
62 std::unique_ptr<HloComputation> MakeCallingComputation( in MakeCallingComputation()
63 HloComputation* callee_computation, int64 callsites, in MakeCallingComputation()
65 HloComputation::Builder builder(TestName() + suffix); in MakeCallingComputation()
78 std::unique_ptr<HloComputation> MakeConditionComputation() { in MakeConditionComputation()
79 HloComputation::Builder builder(TestName() + ".ConditionComputation"); in MakeConditionComputation()
[all …]
|
| D | call_graph_test.cc | 37 std::unique_ptr<HloComputation> MakeScalarComputation( in MakeScalarComputation()
39 HloComputation::Builder builder(TestName() + ".ScalarComputation"); in MakeScalarComputation()
49 std::unique_ptr<HloComputation> MakeMappingComputation( in MakeMappingComputation()
50 HloComputation* map_computation, int64 callsites) { in MakeMappingComputation()
51 HloComputation::Builder builder(TestName() + ".MappingComputation"); in MakeMappingComputation()
64 std::unique_ptr<HloComputation> MakeCallingComputation( in MakeCallingComputation()
65 HloComputation* callee_computation, int64 callsites, in MakeCallingComputation()
67 HloComputation::Builder builder(TestName() + suffix); in MakeCallingComputation()
80 std::unique_ptr<HloComputation> MakeConditionComputation() { in MakeConditionComputation()
81 HloComputation::Builder builder(TestName() + ".ConditionComputation"); in MakeConditionComputation()
[all …]
|
| D | hlo_rematerialization_test.cc | 56 std::unique_ptr<HloComputation> MakeRematerializableComputation( in MakeRematerializableComputation()
58 auto builder = HloComputation::Builder(TestName() + suffix); in MakeRematerializableComputation()
98 std::unique_ptr<HloComputation> MakeRematerializableWhileComputation( in MakeRematerializableWhileComputation()
99 HloComputation* while_cond, HloComputation* while_body, in MakeRematerializableWhileComputation()
101 auto builder = HloComputation::Builder(TestName() + suffix); in MakeRematerializableWhileComputation()
124 std::unique_ptr<HloComputation> MakeConditionComputation() { in MakeConditionComputation()
125 auto builder = HloComputation::Builder(TestName() + ".cond"); in MakeConditionComputation()
148 HloComputation* computation = in TEST_F()
188 HloComputation* computation = in TEST_F()
214 auto cond_builder = HloComputation::Builder(TestName() + ".cond"); in TEST_F()
[all …]
|
| D | while_loop_invariant_code_motion_test.cc | 32 HloComputation* MakeAlwaysTrueComputation(const Shape& param_shape,
36 static void FindOnlyWhileInstruction(HloComputation* computation, in FindOnlyWhileInstruction()
49 HloComputation* WhileLoopInvariantCodeMotionTest::MakeAlwaysTrueComputation( in MakeAlwaysTrueComputation()
51 HloComputation::Builder builder(TestName() + ".always_true"); in MakeAlwaysTrueComputation()
64 HloComputation* while_body = [&]() { in TEST_F()
65 HloComputation::Builder builder(TestName() + ".while_body"); in TEST_F()
81 HloComputation::Builder builder(TestName()); in TEST_F()
87 HloComputation* entry_computation = in TEST_F()
106 HloComputation* while_body = [&]() { in TEST_F()
107 HloComputation::Builder builder(TestName() + ".while_body"); in TEST_F()
[all …]
|
| D | call_graph.cc | 76 [](string* out, const HloComputation* computation) { in ToString()
81 CallGraphNode::CallGraphNode(HloComputation* computation) in CallGraphNode()
95 HloComputation* caller = caller_callsite.instruction()->parent(); in AddCallerCallSite()
125 const HloComputation* computation) const { in GetNode()
131 CallGraphNode& CallGraph::GetNode(const HloComputation* computation) { in GetNode()
138 const HloComputation* a, const HloComputation* b, in DominatesHelper()
139 tensorflow::gtl::FlatSet<const HloComputation*>* visited) const { in DominatesHelper()
154 for (const HloComputation* b_caller : b_node.callers()) { in DominatesHelper()
162 bool CallGraph::Dominates(const HloComputation* a, in Dominates()
163 const HloComputation* b) const { in Dominates()
[all …]
|
| D | hlo_alias_analysis_test.cc | 117 auto builder = HloComputation::Builder(TestName()); in TEST_F()
145 auto builder = HloComputation::Builder(TestName()); in TEST_F()
193 auto builder = HloComputation::Builder(TestName()); in TEST_F()
219 auto subbuilder = HloComputation::Builder("Subcomputation"); in TEST_F()
226 HloComputation* called_computation = in TEST_F()
229 auto builder = HloComputation::Builder(TestName()); in TEST_F()
258 auto subbuilder = HloComputation::Builder("Subcomputation"); in TEST_F()
265 HloComputation* called_computation = in TEST_F()
268 auto builder = HloComputation::Builder(TestName()); in TEST_F()
331 auto body_builder = HloComputation::Builder("body"); in TEST_F()
[all …]
|
| D | call_inliner_test.cc | 49 HloComputation::Builder inner(TestName() + ".inner"); in TEST_F()
56 HloComputation* inner_computation = in TEST_F()
60 HloComputation::Builder outer(TestName() + ".outer"); in TEST_F()
88 HloComputation::Builder just_false(TestName() + ".false"); in TEST_F()
91 HloComputation* false_computation = in TEST_F()
94 HloComputation::Builder call_false_builder(TestName() + ".call_false"); in TEST_F()
97 HloComputation* call_false = in TEST_F()
100 HloComputation::Builder outer(TestName() + ".outer"); in TEST_F()
124 HloComputation::Builder just_false(TestName() + ".false"); in TEST_F()
130 HloComputation* false_computation = in TEST_F()
[all …]
|
| D | hlo_instruction.h | 57 class HloComputation; variable
193 const tensorflow::gtl::FlatMap<string, HloComputation*>& computation_map,
194 const std::function<void(std::unique_ptr<HloComputation>)>&
252 HloComputation* map_computation,
371 HloComputation* reduce_computation);
378 const Window& window, HloComputation* reduce_computation);
400 const Shape& shape, HloInstruction* operand, HloComputation* select,
402 HloComputation* scatter);
445 HloComputation* condition,
446 HloComputation* body,
[all …]
|
| D | copy_insertion_test.cc | 42 int64 CountCopies(const HloComputation& computation) { in CountCopies()
60 int64 CountControlEdges(const HloComputation& computation) { in CountControlEdges()
89 auto builder = HloComputation::Builder(TestName()); in TEST_F()
109 auto builder = HloComputation::Builder(TestName()); in TEST_F()
132 auto builder = HloComputation::Builder(TestName()); in TEST_F()
162 auto builder = HloComputation::Builder(TestName()); in TEST_F()
193 auto builder = HloComputation::Builder(TestName()); in TEST_F()
230 auto builder = HloComputation::Builder(TestName()); in TEST_F()
252 auto builder = HloComputation::Builder(TestName()); in TEST_F()
273 auto builder = HloComputation::Builder(TestName()); in TEST_F()
[all …]
|
| D | hlo_ordering_test.cc | 49 auto builder = HloComputation::Builder(TestName()); in TEST_F()
105 auto builder_c = HloComputation::Builder("C"); in TEST_F()
108 HloComputation* computation_c = in TEST_F()
111 auto builder_b = HloComputation::Builder("B"); in TEST_F()
116 HloComputation* computation_b = in TEST_F()
119 auto builder_a = HloComputation::Builder("A"); in TEST_F()
122 HloComputation* computation_a = in TEST_F()
125 auto builder = HloComputation::Builder(TestName()); in TEST_F()
178 auto body_builder = HloComputation::Builder("body"); in TEST_F()
183 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
[all …]
|
| D | hlo_dataflow_analysis_test.cc | 81 std::unique_ptr<HloComputation> CreateR0F32UnaryOpComputation( in CreateR0F32UnaryOpComputation()
83 HloComputation::Builder builder(TestName() + "." + HloOpcodeString(opcode)); in CreateR0F32UnaryOpComputation()
102 auto builder = HloComputation::Builder(TestName()); in TEST_P()
144 auto builder = HloComputation::Builder(TestName()); in TEST_P()
199 auto builder = HloComputation::Builder(TestName()); in TEST_P()
250 auto subbuilder = HloComputation::Builder("Subcomputation"); in TEST_P()
257 HloComputation* called_computation = in TEST_P()
260 auto builder = HloComputation::Builder(TestName()); in TEST_P()
299 auto subbuilder = HloComputation::Builder("Subcomputation"); in TEST_P()
306 HloComputation* called_computation = in TEST_P()
[all …]
|
| D | while_util.cc | 22 static StatusOr<HloComputation*> WidenWhileCondition( in WidenWhileCondition()
23 HloComputation* narrow_condition, const Shape& wide_shape) { in WidenWhileCondition()
27 HloComputation* wide_while_cond = [&]() { in WidenWhileCondition()
28 HloComputation::Builder builder( in WidenWhileCondition()
55 static StatusOr<std::pair<HloComputation*, CallInliner::InlinedInstructionMap>>
56 WidenWhileBody(HloComputation* narrow_body, const Shape& wide_shape) { in WidenWhileBody()
59 HloComputation* wide_while_body = [&]() { in WidenWhileBody()
60 HloComputation::Builder builder( in WidenWhileBody()
103 HloComputation * new_while_condition, in MakeInstructionsLiveIn()
106 HloComputation* new_while_body; in MakeInstructionsLiveIn()
[all …]
|
| D | layout_assignment_test.cc | 66 auto builder = HloComputation::Builder(TestName()); in TEST_F()
75 HloComputation* computation = module->AddEntryComputation(builder.Build()); in TEST_F()
96 auto builder = HloComputation::Builder(TestName()); in TEST_F()
105 HloComputation* computation = module->AddEntryComputation(builder.Build()); in TEST_F()
135 auto builder = HloComputation::Builder(TestName()); in TEST_F()
154 HloComputation* computation = module->AddEntryComputation(builder.Build()); in TEST_F()
185 auto builder = HloComputation::Builder(TestName()); in TEST_F()
222 auto builder = HloComputation::Builder(TestName()); in TEST_F()
267 auto builder = HloComputation::Builder(TestName()); in TEST_F()
322 auto builder = HloComputation::Builder(TestName()); in TEST_F()
[all …]
|
| D | bfloat16_normalization.cc | 31 explicit BFloat16NormalizationVisitor(HloComputation* computation, in BFloat16NormalizationVisitor()
40 static bool Run(HloComputation* computation, in Run()
54 HloComputation* computation);
60 HloComputation* computation);
65 HloComputation* computation);
71 tensorflow::gtl::ArraySlice<HloComputation*> bf16_called_comps);
73 HloComputation* computation_;
79 HloInstruction* hlo, PrimitiveType to, HloComputation* computation) { in InsertConvertAfterOutput()
97 HloInstruction* hlo, PrimitiveType to, HloComputation* computation) { in ChangeOutputTypeThenInsertConvertBack()
105 HloComputation* computation) { in InsertConvertBeforeOperand()
[all …]
|
| D | hlo_computation.h | 55 class HloComputation {
70 std::unique_ptr<HloComputation> Build(
171 static StatusOr<std::unique_ptr<HloComputation>> CreateFromProto(
173 const tensorflow::gtl::FlatMap<string, HloComputation*>& computation_map,
174 const std::function<void(std::unique_ptr<HloComputation>)>&
222 std::list<HloComputation*> MakeEmbeddedComputationsList() const;
255 bool operator==(const HloComputation& other) const;
305 std::unique_ptr<HloComputation> Clone(const string& suffix = "clone",
313 std::unique_ptr<HloComputation> CloneWithReplacements(
346 explicit HloComputation(
[all …]
|
| D | liveness_util_test.cc | 29 void BuildModule(std::unique_ptr<HloComputation> computation) { in BuildModule()
41 void BuildModuleAndRunAnalysis(std::unique_ptr<HloComputation> computation) { in BuildModuleAndRunAnalysis()
47 HloComputation* computation_ = nullptr;
55 auto builder = HloComputation::Builder(TestName()); in TEST_F()
83 auto builder = HloComputation::Builder(TestName()); in TEST_F()
124 auto builder = HloComputation::Builder(TestName()); in TEST_F()
148 auto builder = HloComputation::Builder(TestName()); in TEST_F()
173 auto builder = HloComputation::Builder(TestName()); in TEST_F()
197 auto builder = HloComputation::Builder(TestName()); in TEST_F()
237 auto builder = HloComputation::Builder(TestName()); in TEST_F()
[all …]
|
| D | hlo_dce_test.cc | 44 bool HasInstruction(const HloComputation& computation, in HasInstruction()
54 auto builder = HloComputation::Builder(TestName()); in TEST_F()
75 auto builder = HloComputation::Builder(TestName()); in TEST_F()
96 auto builder = HloComputation::Builder(TestName()); in TEST_F()
127 auto builder = HloComputation::Builder(TestName()); in TEST_F()
181 auto callee_builder = HloComputation::Builder(TestName() + "-callee"); in TEST_F()
192 auto builder = HloComputation::Builder(TestName()); in TEST_F()
221 auto cond_builder = HloComputation::Builder(TestName() + "-cond"); in TEST_F()
233 auto body_builder = HloComputation::Builder(TestName() + "-body"); in TEST_F()
246 auto builder = HloComputation::Builder(TestName()); in TEST_F()
[all …]
|
| D | implicit_broadcast_remover_test.cc | 34 auto builder = HloComputation::Builder(TestName()); in TEST_F()
44 HloComputation* computation = module().AddEntryComputation(builder.Build()); in TEST_F()
53 auto builder = HloComputation::Builder(TestName()); in TEST_F()
63 HloComputation* computation = module().AddEntryComputation(builder.Build()); in TEST_F()
79 auto builder = HloComputation::Builder(TestName()); in TEST_F()
89 HloComputation* computation = module().AddEntryComputation(builder.Build()); in TEST_F()
101 auto builder = HloComputation::Builder(TestName()); in TEST_F()
111 HloComputation* computation = module().AddEntryComputation(builder.Build()); in TEST_F()
123 auto builder = HloComputation::Builder(TestName()); in TEST_F()
135 HloComputation* computation = module().AddEntryComputation(builder.Build()); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_convolution_rewriter_test.cc | 96 HloComputation::Builder builder(TestName()); in TEST_F()
115 HloComputation* entry_computation = in TEST_F()
125 HloComputation::Builder builder(TestName()); in TEST_F()
143 HloComputation* entry_computation = in TEST_F()
154 auto builder = HloComputation::Builder(TestName()); in TEST_F()
173 HloComputation* entry_computation = in TEST_F()
184 auto builder = HloComputation::Builder(TestName()); in TEST_F()
203 HloComputation* entry_computation = in TEST_F()
212 auto builder = HloComputation::Builder(TestName()); in TEST_F()
231 HloComputation* entry_computation = in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_copy_insertion_test.cc | 37 int64 CountCopies(const HloComputation& computation) { in CountCopies()
69 auto builder = HloComputation::Builder(TestName()); in TEST_F()
73 auto body_builder = HloComputation::Builder("body"); in TEST_F()
78 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
80 auto cond_builder = HloComputation::Builder("condition"); in TEST_F()
85 HloComputation* condition = in TEST_F()
107 auto builder = HloComputation::Builder(TestName()); in TEST_F()
113 auto sub_builder = HloComputation::Builder("subcomputation"); in TEST_F()
122 HloComputation* subcomputation = in TEST_F()
|