Home
last modified time | relevance | path

Searched refs:root_instruction (Results 1 – 25 of 226) sorted by relevance

12345678910

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible_test.cc47 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
79 module->entry_computation()->root_instruction()->operand(1); in TEST_F()
112 module->entry_computation()->root_instruction()->operand(1); in TEST_F()
132 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
152 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
183 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
208 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
224 module->entry_computation()->root_instruction(); in TEST_F()
242 module->entry_computation()->root_instruction(); in TEST_F()
261 module->entry_computation()->root_instruction(); in TEST_F()
[all …]
Dinstruction_fusion_test.cc47 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()
51 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()
67 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()
71 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F()
86 EXPECT_EQ(reshape2, computation->root_instruction()); in TEST_F()
90 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F()
105 EXPECT_EQ(transpose2, computation->root_instruction()); in TEST_F()
109 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F()
125 EXPECT_EQ(log, computation->root_instruction()); in TEST_F()
142 EXPECT_EQ(transpose2, computation->root_instruction()); in TEST_F()
[all …]
Dcudnn_pad_for_convolutions_test.cc47 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
72 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
94 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()
161 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
182 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
206 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
230 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
251 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dar_crs_combiner_test.cc44 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
64 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
83 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
104 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
124 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
144 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
166 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
188 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
211 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
247 auto root_while = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
Droot_instruction_sinker_test.cc58 module->entry_computation()->root_instruction()->while_body(); in TEST_F()
63 ->root_instruction() in TEST_F()
104 module->entry_computation()->root_instruction()->while_body(); in TEST_F()
107 while_body->root_instruction()); in TEST_F()
108 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
129 module->entry_computation()->root_instruction()->called_computations()[0]; in TEST_F()
134 ->root_instruction() in TEST_F()
161 module->entry_computation()->root_instruction()->called_computations()[0]; in TEST_F()
164 called_computation->root_instruction()); in TEST_F()
165 EXPECT_THAT(called_computation->root_instruction(), in TEST_F()
Dreshape_mover_test.cc57 EXPECT_THAT(computation->root_instruction(), in TEST_F()
62 EXPECT_THAT(computation->root_instruction(), in TEST_F()
100 EXPECT_THAT(computation->root_instruction(),
105 EXPECT_THAT(computation->root_instruction(),
126 EXPECT_THAT(computation->root_instruction(), in TEST_F()
132 computation->root_instruction(), in TEST_F()
153 EXPECT_THAT(computation->root_instruction(), in TEST_F()
157 EXPECT_THAT(computation->root_instruction(), in TEST_F()
160 computation->root_instruction()->shape().DebugString()); in TEST_F()
207 EXPECT_THAT(computation->root_instruction(),
[all …]
Dstable_sort_expander_test.cc107 auto root = module->entry_computation()->root_instruction(); in TEST_F()
111 root->operand(0)->to_apply()->root_instruction(), /*iota_parameter=*/1); in TEST_F()
153 auto root = module->entry_computation()->root_instruction(); in TEST_F()
157 root->operand(0)->to_apply()->root_instruction(), /*iota_parameter=*/1); in TEST_F()
183 auto root = module->entry_computation()->root_instruction(); in TEST_F()
191 root->operand(0)->operand(0)->to_apply()->root_instruction(), in TEST_F()
251 auto root = module->entry_computation()->root_instruction(); in TEST_F()
255 root->operand(0)->to_apply()->root_instruction(), in TEST_F()
288 auto root = module->entry_computation()->root_instruction(); in TEST_F()
292 root->operand(0)->to_apply()->root_instruction(), in TEST_F()
[all …]
Dalgebraic_simplifier_test.cc71 HloInstruction* root = computation->root_instruction(); in TEST_F()
75 root = computation->root_instruction(); in TEST_F()
95 m->entry_computation()->root_instruction(), in TEST_F()
115 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
133 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
154 m->entry_computation()->root_instruction(), in TEST_F()
174 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
197 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
219 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
277 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
[all …]
Dwhile_loop_simplifier_test.cc132 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()
168 auto* while_op = computation->root_instruction(); in TEST_F()
173 while_op->while_body()->root_instruction())); in TEST_F()
175 EXPECT_THAT(computation->root_instruction()->control_predecessors(), in TEST_F()
185 auto* while_op = computation->root_instruction(); in TEST_F()
201 auto* while_op = computation->root_instruction(); in TEST_F()
217 auto* while_op = computation->root_instruction(); in TEST_F()
[all …]
Dhlo_cost_analysis_test.cc149 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
160 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
183 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
194 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
219 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
230 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
249 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
260 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
276 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
283 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
[all …]
Dhlo_constant_folding_test.cc55 EXPECT_THAT(computation->root_instruction(), in TEST_F()
62 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F()
64 computation->root_instruction()->literal().GetFirstElement<int64_t>(), in TEST_F()
78 EXPECT_THAT(computation->root_instruction(), in TEST_F()
85 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F()
86 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<float>(), in TEST_F()
100 EXPECT_THAT(computation->root_instruction(), in TEST_F()
107 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F()
108 EXPECT_EQ(computation->root_instruction()->literal().Get<int64_t>({0}), 42); in TEST_F()
109 EXPECT_EQ(computation->root_instruction()->literal().Get<int64_t>({1}), 19); in TEST_F()
[all …]
Dtopk_rewriter_test.cc139 module->entry_computation()->root_instruction(), in TEST_F()
143 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F()
170 module->entry_computation()->root_instruction(), in TEST_F()
174 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F()
201 module->entry_computation()->root_instruction(), in TEST_F()
205 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F()
233 module->entry_computation()->root_instruction(), in TEST_F()
239 ->root_instruction() in TEST_F()
262 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
265 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
Dconvolution_4d_expander_test.cc45 HloInstruction* root = computation->root_instruction(); in TEST_F()
50 root = computation->root_instruction(); in TEST_F()
70 HloInstruction* root = computation->root_instruction(); in TEST_F()
75 root = computation->root_instruction(); in TEST_F()
97 HloInstruction* root = computation->root_instruction(); in TEST_F()
102 root = computation->root_instruction(); in TEST_F()
122 HloInstruction* root = computation->root_instruction(); in TEST_F()
141 HloInstruction* root = computation->root_instruction(); in TEST_F()
160 HloInstruction* root = computation->root_instruction(); in TEST_F()
Dwhile_loop_constant_sinking_test.cc64 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
111 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
151 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
192 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
240 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
282 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
328 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F()
368 EXPECT_THAT(while_condition->root_instruction(), in TEST_F()
411 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F()
461 EXPECT_THAT(while_condition->root_instruction(), in TEST_F()
Dlogistic_expander_test.cc64 HloInstruction* root = computation->root_instruction(); in TEST_F()
68 root = computation->root_instruction(); in TEST_F()
69 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
91 HloInstruction* root = computation->root_instruction(); in TEST_F()
95 root = computation->root_instruction(); in TEST_F()
96 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
Dcall_inliner_test.cc68 EXPECT_THAT(computation->root_instruction(), op::Constant()); in TEST_F()
69 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<float>(), in TEST_F()
71 ASSERT_EQ(1, computation->root_instruction()->control_predecessors().size()); in TEST_F()
72 auto prior = computation->root_instruction()->control_predecessors()[0]; in TEST_F()
112 computation->root_instruction()->while_condition()->root_instruction(), in TEST_F()
114 EXPECT_THAT(computation->root_instruction()->while_body()->root_instruction(), in TEST_F()
139 EXPECT_THAT(computation->root_instruction(), op::Constant()); in TEST_F()
140 EXPECT_THAT(computation->root_instruction()->control_successors(), in TEST_F()
181 EXPECT_THAT(computation->root_instruction(), op::Constant()); in TEST_F()
199 outfeed_computation->root_instruction()->shape(), /*operands=*/{}, in TEST_F()
Dwhile_loop_analysis_test.cc55 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()
Dinstruction_fusion_test.cc57 HloInstruction* sub = module->entry_computation()->root_instruction(); in TEST_F()
81 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
101 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
130 EXPECT_EQ(unary, computation->root_instruction()); in TEST_F()
196 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
302 root = module->entry_computation()->root_instruction(); in TEST_F()
329 EXPECT_EQ(unary2, computation->root_instruction()); in TEST_F()
357 EXPECT_EQ(unary, computation->root_instruction()); in TEST_F()
388 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
417 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
Dconditional_code_motion_test.cc78 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
163 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
204 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
289 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
358 for (int i = 0; i < on_false->root_instruction()->operand_count(); ++i) { in TEST_F()
360 on_false->root_instruction()->operand(i); in TEST_F()
370 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
424 for (int i = 0; i < on_false->root_instruction()->operand_count(); ++i) { in TEST_F()
426 on_false->root_instruction()->operand(i); in TEST_F()
435 EXPECT_THAT(on_false->root_instruction()->operand(*on_false_const_idx), in TEST_F()
[all …]
Dconvolution_group_converter_test.cc49 HloInstruction* root = computation->root_instruction(); in TEST_F()
56 root = computation->root_instruction(); in TEST_F()
82 HloInstruction* root = computation->root_instruction(); in TEST_F()
90 root = computation->root_instruction(); in TEST_F()
111 HloInstruction* root = computation->root_instruction(); in TEST_F()
119 root = computation->root_instruction(); in TEST_F()
140 HloInstruction* root = computation->root_instruction(); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_control_flow_flattening_test.cc70 auto root = module->entry_computation()->root_instruction(); in TEST_F()
79 condition->root_instruction(), in TEST_F()
83 EXPECT_THAT(body->root_instruction(), in TEST_F()
131 auto root = module->entry_computation()->root_instruction(); in TEST_F()
140 condition->root_instruction(), in TEST_F()
144 EXPECT_THAT(body->root_instruction(), in TEST_F()
185 auto root = module->entry_computation()->root_instruction(); in TEST_F()
194 condition->root_instruction(), in TEST_F()
241 auto fusion = module->entry_computation()->root_instruction(); in TEST_F()
264 auto tuple = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
Dhlo_extractor_test.cc47 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
54 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
61 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
85 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
92 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
98 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
105 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
129 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
135 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_reduce_scatter_creator_test.cc95 ASSERT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
98 module->entry_computation()->root_instruction()); in TEST_F()
134 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
167 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
201 ASSERT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
204 module->entry_computation()->root_instruction()); in TEST_F()
269 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
304 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
338 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
377 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc110 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
134 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
152 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
173 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
197 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
219 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
242 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
262 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
286 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
316 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
Dschedule_aware_collective_ops_cse_test.cc60 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F()
81 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F()
104 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F()
127 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F()
150 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F()
171 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F()
194 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F()
215 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F()
236 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F()

12345678910