| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils_test.cc | 33 HloComputation** entry_computation) { in CreateModuleWithProgramShape() argument
38 *entry_computation = module->AddEntryComputation( in CreateModuleWithProgramShape()
41 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape()
48 HloComputation** entry_computation, PrimitiveType primitive_type_output) { in CreateModuleWithProgramShape() argument
53 *entry_computation = module->AddEntryComputation( in CreateModuleWithProgramShape()
56 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape()
63 HloComputation* entry_computation; in TEST_F() local
67 &entry_computation); in TEST_F()
71 entry_computation->set_root_instruction(first_1_dims_collapsed); in TEST_F()
82 HloComputation* entry_computation; in TEST_F() local
[all …]
|
| D | hlo_module_dce_test.cc | 113 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
115 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
153 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
155 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
191 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
194 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
197 auto while_loops = GetWhileLoops(module->entry_computation()); in TEST_F()
235 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
238 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
241 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
[all …]
|
| D | ar_crs_combiner_test.cc | 43 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
47 i1, module->entry_computation()->parameter_instruction(0))); in TEST_F()
63 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
82 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
103 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
123 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
143 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
165 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
187 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
210 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
|
| D | while_util_test.cc | 35 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule() argument
60 *entry_computation = module->entry_computation(); in GetParsedModule()
61 *param0 = (*entry_computation)->parameter_instruction(0); in GetParsedModule()
62 *param1 = (*entry_computation)->parameter_instruction(1); in GetParsedModule()
63 *param2 = (*entry_computation)->parameter_instruction(2); in GetParsedModule()
71 HloComputation* entry_computation; in TEST_F() local
75 GetParsedModule(&entry_computation, ¶m0, ¶m1, ¶m2)); in TEST_F()
77 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST_F()
87 entry_computation->root_instruction(), in TEST_F()
102 HloComputation* entry_computation; in TEST_F() local
[all …]
|
| D | while_loop_all_reduce_code_motion_test.cc | 94 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F()
95 module->entry_computation()->instructions().end(), in TEST_F()
107 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F()
108 module->entry_computation()->instructions().end(), in TEST_F()
193 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F()
194 module->entry_computation()->instructions().end(), in TEST_F()
203 absl::c_copy_if(module->entry_computation()->instructions(), in TEST_F()
272 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F()
273 module->entry_computation()->instructions().end(), in TEST_F()
281 HloInstruction* new_root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
|
| D | tuple_util_test.cc | 34 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule() argument
51 *entry_computation = module->entry_computation(); in GetParsedModule()
52 *param0 = (*entry_computation)->parameter_instruction(0); in GetParsedModule()
53 *param1 = (*entry_computation)->parameter_instruction(1); in GetParsedModule()
60 HloComputation* entry_computation; in TEST() local
63 auto module, GetParsedModule(&entry_computation, ¶m0, ¶m1)); in TEST()
73 HloComputation* entry_computation; in TEST() local
76 auto module, GetParsedModule(&entry_computation, ¶m0, ¶m1)); in TEST()
|
| D | call_graph_test.cc | 118 HloComputation* entry_computation = in TEST_F() local
126 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation); in TEST_F()
128 EXPECT_EQ(entry_computation, entry_node.computation()); in TEST_F()
144 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local
150 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation); in TEST_F()
151 EXPECT_EQ(entry_computation, entry_node.computation()); in TEST_F()
175 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local
185 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation); in TEST_F()
186 EXPECT_EQ(entry_computation, entry_node.computation()); in TEST_F()
216 HloComputation* entry_computation = in TEST_F() local
[all …]
|
| D | hlo_rematerialization_test.cc | 160 HloComputation* entry_computation = in TEST_F() local
164 EXPECT_EQ(entry_computation->instruction_count(), 7); in TEST_F()
176 EXPECT_EQ(entry_computation->instruction_count(), 8); in TEST_F()
196 HloComputation* entry_computation = in TEST_F() local
200 EXPECT_EQ(entry_computation->instruction_count(), 7); in TEST_F()
209 EXPECT_EQ(entry_computation->instruction_count(), 9); in TEST_F()
234 HloComputation* entry_computation = in TEST_F() local
238 EXPECT_EQ(entry_computation->instruction_count(), 7); in TEST_F()
250 EXPECT_EQ(entry_computation->instruction_count(), 9); in TEST_F()
292 HloComputation* entry_computation = in TEST_F() local
[all …]
|
| D | hlo_cost_analysis_test.cc | 148 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
159 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
182 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
193 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
218 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
229 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
248 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
259 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
275 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
282 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
[all …]
|
| D | while_loop_analysis_test.cc | 55 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F()
86 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F()
119 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F()
154 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F()
193 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F()
232 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F()
267 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F()
|
| D | topk_rewriter_test.cc | 85 module->entry_computation()->root_instruction(), in TEST_F()
89 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F()
115 module->entry_computation()->root_instruction(), in TEST_F()
119 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F()
144 LOG(INFO) << module->entry_computation()->ToString(); in TEST_F()
146 module->entry_computation()->root_instruction(), in TEST_F()
151 const HloInstruction* cc = module->entry_computation() in TEST_F()
|
| D | transpose_folding_test.cc | 77 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
155 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
177 HloComputation* entry_computation = in TEST_F() local
180 {add, sub, mul}, "entry", entry_computation); in TEST_F()
181 EXPECT_EQ(call, entry_computation->root_instruction()); in TEST_F()
254 HloComputation* entry_computation = in TEST_F() local
260 entry_computation->instructions().begin(), in TEST_F()
261 entry_computation->instructions().end()); in TEST_F()
311 HloComputation* entry_computation = in TEST_F() local
317 entry_computation->instructions().begin(), in TEST_F()
[all …]
|
| D | root_instruction_sinker_test.cc | 58 module->entry_computation()->root_instruction()->while_body(); in TEST_F()
62 EXPECT_EQ(module->entry_computation() in TEST_F()
104 module->entry_computation()->root_instruction()->while_body(); in TEST_F()
129 module->entry_computation()->root_instruction()->called_computations()[0]; in TEST_F()
133 EXPECT_EQ(module->entry_computation() in TEST_F()
161 module->entry_computation()->root_instruction()->called_computations()[0]; in TEST_F()
|
| D | while_loop_simplifier_test.cc | 39 const auto& instrs = m->entry_computation()->instructions(); in FindFirstWhile()
132 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
140 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
147 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
155 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
167 HloComputation* computation = m->entry_computation(); in TEST_F()
184 HloComputation* computation = m->entry_computation(); in TEST_F()
200 HloComputation* computation = m->entry_computation(); in TEST_F()
216 HloComputation* computation = m->entry_computation(); in TEST_F()
231 HloComputation* computation = m->entry_computation(); in TEST_F()
[all …]
|
| D | hlo_memory_scheduler_test.cc | 51 HloComputation* computation = module->entry_computation(); in PeakMemoryUseOfEntryComputation()
101 module->schedule().sequence(module->entry_computation()).instructions(); in TEST_F()
102 EXPECT_EQ(module->entry_computation()->instruction_count(), sequence.size()); in TEST_F()
153 schedule.sequence(module->entry_computation()).instructions(); in TEST_F()
154 EXPECT_EQ(module->entry_computation()->instruction_count(), sequence.size()); in TEST_F()
204 schedule.sequence(module->entry_computation()).instructions(); in TEST_F()
205 EXPECT_EQ(module->entry_computation()->instruction_count(), sequence.size()); in TEST_F()
253 EXPECT_EQ(module->entry_computation()->instruction_count(), in TEST_F()
254 schedule.sequence(module->entry_computation()).size()); in TEST_F()
304 EXPECT_EQ(module->entry_computation()->instruction_count(), in TEST_F()
[all …]
|
| D | conditional_code_motion_test.cc | 77 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
162 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
203 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
288 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
356 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
411 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
617 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
704 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
749 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
788 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
|
| D | convolution_4d_expander_test.cc | 44 auto computation = module->entry_computation(); in TEST_F()
69 auto computation = module->entry_computation(); in TEST_F()
96 auto computation = module->entry_computation(); in TEST_F()
121 auto computation = module->entry_computation(); in TEST_F()
140 auto computation = module->entry_computation(); in TEST_F()
159 auto computation = module->entry_computation(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible_test.cc | 48 module->entry_computation()->root_instruction(); in TEST_F()
51 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
84 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
88 module->entry_computation()->root_instruction()->operand(1); in TEST_F()
122 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
126 module->entry_computation()->root_instruction()->operand(1); in TEST_F()
147 module->entry_computation()->root_instruction(); in TEST_F()
150 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
182 module->entry_computation()->root_instruction(); in TEST_F()
186 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
[all …]
|
| D | gpu_conv_rewriter_test.cc | 125 HloComputation* entry_computation = in TEST_F() local
128 ASSERT_THAT(entry_computation->root_instruction(), in TEST_F()
134 entry_computation->root_instruction()->operand(0)->metadata(); in TEST_F()
162 HloComputation* entry_computation = in TEST_F() local
166 entry_computation->root_instruction(), in TEST_F()
192 HloComputation* entry_computation = in TEST_F() local
195 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
222 HloComputation* entry_computation = in TEST_F() local
225 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
251 HloComputation* entry_computation = in TEST_F() local
[all …]
|
| D | gpu_hlo_schedule.cc | 47 return &computation == module_->entry_computation() ? entry_sequence_.get() in SequentialOrder()
80 module->entry_computation()->MakeInstructionPostOrder()); in GpuHloOrdering()
114 predecessors_.emplace(module->entry_computation(), in GpuHloOrdering()
125 if (computation != module->entry_computation() && in GpuHloOrdering()
198 HloComputation* entry_computation = module.entry_computation(); in Build() local
205 entry_computation, [pointer_size](const BufferValue& buffer) { in Build()
211 BFSLaunchOrder(entry_computation, &schedule->thunk_launch_order_); in Build()
|
| D | cudnn_pad_for_convolutions_test.cc | 47 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
72 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
95 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
116 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
138 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
160 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
181 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
206 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
232 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_extractor_test.cc | 45 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
52 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
59 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
83 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
90 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
96 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
103 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
127 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
133 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 76 auto root = module->entry_computation()->root_instruction(); in TEST_F()
87 auto root = module->entry_computation()->root_instruction(); in TEST_F()
101 auto root = module->entry_computation()->root_instruction(); in TEST_F()
132 auto root = module->entry_computation()->root_instruction(); in TEST_F()
149 module->entry_computation()->root_instruction(), in TEST_F()
165 module->entry_computation()->root_instruction(), in TEST_F()
187 auto root = module->entry_computation()->root_instruction(); in TEST_F()
203 auto root = module->entry_computation()->root_instruction(); in TEST_F()
212 auto root = module->entry_computation()->root_instruction(); in TEST_F()
263 auto root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 99 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
117 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
138 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
162 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
184 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
207 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
227 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
251 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
281 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
308 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
|
| D | canonicalize_all_gather_for_cse_test.cc | 67 module->entry_computation()->root_instruction(); in TEST_F()
87 module->entry_computation()->root_instruction(); in TEST_F()
107 module->entry_computation()->root_instruction(); in TEST_F()
127 module->entry_computation()->root_instruction(); in TEST_F()
147 module->entry_computation()->root_instruction(); in TEST_F()
167 module->entry_computation()->root_instruction(); in TEST_F()
|