Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible_test.cc48 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()
109 module->entry_computation()->root_instruction(); in TEST_F()
112 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
144 module->entry_computation()->root_instruction(); in TEST_F()
148 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
173 module->entry_computation()->root_instruction(); in TEST_F()
176 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
[all …]
Dinstruction_fusion_test.cc46 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()
50 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()
66 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()
70 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F()
85 EXPECT_EQ(reshape2, computation->root_instruction()); in TEST_F()
89 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F()
104 EXPECT_EQ(transpose2, computation->root_instruction()); in TEST_F()
108 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F()
122 EXPECT_EQ(reshape2, computation->root_instruction()); in TEST_F()
139 EXPECT_EQ(transpose2, computation->root_instruction()); in TEST_F()
[all …]
Dcudnn_conv_pad_for_tensor_cores_test.cc47 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()
Dcudnn_conv_rewriter_test.cc127 ASSERT_THAT(entry_computation->root_instruction(), in TEST_F()
133 entry_computation->root_instruction()->operand(0)->metadata(); in TEST_F()
163 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
193 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
223 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
252 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
306 ASSERT_THAT(entry_computation->root_instruction(), in TEST_F()
310 entry_computation->root_instruction()->operand(0); in TEST_F()
353 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
388 entry_computation->root_instruction(), in TEST_F()
[all …]
Dmulti_output_fusion_test.cc70 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F()
158 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F()
191 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F()
253 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F()
297 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F()
353 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F()
399 const HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
426 const HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
466 const HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
533 const HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dar_crs_combiner_test.cc43 auto root_tuple = module->entry_computation()->root_instruction(); 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()
246 auto root_while = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
Dreshape_mover_test.cc56 EXPECT_THAT(computation->root_instruction(), in TEST_F()
61 EXPECT_THAT(computation->root_instruction(), in TEST_F()
99 EXPECT_THAT(computation->root_instruction(),
104 EXPECT_THAT(computation->root_instruction(),
125 EXPECT_THAT(computation->root_instruction(), in TEST_F()
131 computation->root_instruction(), in TEST_F()
152 EXPECT_THAT(computation->root_instruction(), in TEST_F()
156 EXPECT_THAT(computation->root_instruction(), in TEST_F()
159 computation->root_instruction()->shape().DebugString()); in TEST_F()
206 EXPECT_THAT(computation->root_instruction(),
[all …]
Dalgebraic_simplifier_test.cc68 HloInstruction* root = computation->root_instruction(); in TEST_F()
72 root = computation->root_instruction(); in TEST_F()
92 m->entry_computation()->root_instruction(), in TEST_F()
112 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
134 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
192 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
210 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
233 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
250 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
273 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
[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 …]
Dhlo_constant_folding_test.cc53 EXPECT_THAT(computation->root_instruction(), in TEST_F()
60 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F()
61 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<int64>(), in TEST_F()
75 EXPECT_THAT(computation->root_instruction(), in TEST_F()
82 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F()
83 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<float>(), in TEST_F()
97 EXPECT_THAT(computation->root_instruction(), in TEST_F()
104 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F()
105 EXPECT_EQ(computation->root_instruction()->literal().Get<int64>({0}), 42); in TEST_F()
106 EXPECT_EQ(computation->root_instruction()->literal().Get<int64>({1}), 19); in TEST_F()
[all …]
Dcall_inliner_test.cc69 EXPECT_THAT(computation->root_instruction(), op::Constant()); in TEST_F()
70 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<float>(), in TEST_F()
72 ASSERT_EQ(1, computation->root_instruction()->control_predecessors().size()); in TEST_F()
73 auto prior = computation->root_instruction()->control_predecessors()[0]; in TEST_F()
113 computation->root_instruction()->while_condition()->root_instruction(), in TEST_F()
115 EXPECT_THAT(computation->root_instruction()->while_body()->root_instruction(), in TEST_F()
140 EXPECT_THAT(computation->root_instruction(), op::Constant()); in TEST_F()
141 EXPECT_THAT(computation->root_instruction()->control_successors(), in TEST_F()
160 outfeed_computation->root_instruction()->shape(), /*operands=*/{}, in TEST_F()
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()
219 auto* while_op = computation->root_instruction(); in TEST_F()
[all …]
Dreduce_precision_insertion_test.cc61 EXPECT_EQ(computation->root_instruction(), b); in TEST_F()
70 EXPECT_EQ(computation->root_instruction(), b); in TEST_F()
88 EXPECT_EQ(computation->root_instruction(), b); in TEST_F()
97 EXPECT_EQ(computation->root_instruction(), b); in TEST_F()
118 EXPECT_EQ(computation->root_instruction(), c); in TEST_F()
128 EXPECT_EQ(computation->root_instruction(), c); in TEST_F()
147 EXPECT_EQ(computation->root_instruction(), b); in TEST_F()
157 EXPECT_EQ(computation->root_instruction(), b); in TEST_F()
180 EXPECT_EQ(computation->root_instruction(), d); in TEST_F()
212 EXPECT_EQ(computation->root_instruction(), b); in TEST_F()
[all …]
Dwhile_loop_constant_sinking_test.cc65 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
105 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
146 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
194 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
236 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
282 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F()
322 EXPECT_THAT(while_condition->root_instruction(), in TEST_F()
365 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F()
415 EXPECT_THAT(while_condition->root_instruction(), in TEST_F()
Dhlo_cost_analysis_test.cc149 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
178 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
207 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
230 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
251 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
277 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
306 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
327 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
345 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
364 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
[all …]
Dtuple_simplifier_test.cc104 EXPECT_THAT(computation->root_instruction(), gte); in TEST_F()
108 EXPECT_THAT(computation->root_instruction(), param1); in TEST_F()
131 EXPECT_THAT(computation->root_instruction(), in TEST_F()
136 EXPECT_THAT(computation->root_instruction(), op::Negate(op::Parameter())); in TEST_F()
163 EXPECT_THAT(computation->root_instruction(), element); in TEST_F()
167 EXPECT_THAT(computation->root_instruction(), param); in TEST_F()
188 EXPECT_THAT(computation->root_instruction(), tuple); in TEST_F()
192 EXPECT_THAT(computation->root_instruction(), tuple_param); in TEST_F()
213 EXPECT_THAT(computation->root_instruction(), tuple); in TEST_F()
217 EXPECT_THAT(computation->root_instruction(), tuple); in TEST_F()
[all …]
Dinstruction_fusion_test.cc58 HloInstruction* sub = module->entry_computation()->root_instruction(); in TEST_F()
82 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
102 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
122 EXPECT_EQ(reshape1, computation->root_instruction()); in TEST_F()
138 EXPECT_EQ(reshape1, computation->root_instruction()); in TEST_F()
154 EXPECT_EQ(transpose1, computation->root_instruction()); in TEST_F()
177 EXPECT_EQ(unary, computation->root_instruction()); in TEST_F()
241 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
341 root = module->entry_computation()->root_instruction(); in TEST_F()
366 EXPECT_EQ(unary2, computation->root_instruction()); in TEST_F()
[all …]
Dconvolution_group_converter_test.cc50 HloInstruction* root = computation->root_instruction(); in TEST_F()
55 root = computation->root_instruction(); in TEST_F()
81 HloInstruction* root = computation->root_instruction(); in TEST_F()
86 root = computation->root_instruction(); in TEST_F()
110 HloInstruction* root = computation->root_instruction(); in TEST_F()
116 root = computation->root_instruction(); in TEST_F()
Dwhile_util_test.cc72 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST()
82 entry_computation->root_instruction(), in TEST()
90 EXPECT_THAT(new_while_instr->while_body()->root_instruction(), in TEST()
103 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST()
116 entry_computation->root_instruction(), in TEST()
124 EXPECT_THAT(new_while_instr->while_body()->root_instruction(), in TEST()
199 HloInstruction* while_instr = main->root_instruction(); in TEST()
Ddefuser_test.cc75 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F()
81 EXPECT_THAT(computation->root_instruction(), in TEST_F()
101 EXPECT_THAT(computation->root_instruction(), op::Negate(op::Fusion())); in TEST_F()
107 EXPECT_THAT(computation->root_instruction(), in TEST_F()
140 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F()
146 EXPECT_THAT(computation->root_instruction(), in TEST_F()
180 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F()
186 EXPECT_THAT(computation->root_instruction(), in TEST_F()
210 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F()
216 EXPECT_THAT(computation->root_instruction(), op::Negate(op::Add())); in TEST_F()
Dbatch_dot_simplification_test.cc45 HloInstruction* root = m->entry_computation()->root_instruction(); in TEST_F()
69 HloInstruction* root = m->entry_computation()->root_instruction(); in TEST_F()
93 HloInstruction* root = m->entry_computation()->root_instruction(); in TEST_F()
117 HloInstruction* root = m->entry_computation()->root_instruction(); in TEST_F()
141 HloInstruction* root = m->entry_computation()->root_instruction(); in TEST_F()
165 HloInstruction* root = m->entry_computation()->root_instruction(); in TEST_F()
Ddynamic_index_splitter_test.cc52 ASSERT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
60 ->root_instruction() in TEST_F()
90 ASSERT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
98 ->root_instruction() in TEST_F()
128 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
Dbfloat16_normalization_test.cc115 EXPECT_EQ(computation->root_instruction(), add1); in TEST_F()
143 EXPECT_EQ(computation->root_instruction()->opcode(), HloOpcode::kConvert); in TEST_F()
144 EXPECT_EQ(computation->root_instruction()->operand(0), mul1); in TEST_F()
173 EXPECT_EQ(computation->root_instruction()->opcode(), HloOpcode::kConvert); in TEST_F()
174 EXPECT_EQ(computation->root_instruction()->operand(0), sub1); in TEST_F()
211 EXPECT_EQ(computation->root_instruction(), reduce); in TEST_F()
225 ->root_instruction() in TEST_F()
268 EXPECT_EQ(computation->root_instruction(), gte); in TEST_F()
298 EXPECT_EQ(computation->root_instruction(), gte); in TEST_F()
327 EXPECT_NE(computation->root_instruction(), sort); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor_test.cc45 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/
Dxla_builder_test.cc74 auto root = module->entry_computation()->root_instruction(); in TEST_F()
85 auto root = module->entry_computation()->root_instruction(); in TEST_F()
99 auto root = module->entry_computation()->root_instruction(); in TEST_F()
130 auto root = module->entry_computation()->root_instruction(); in TEST_F()
147 module->entry_computation()->root_instruction(), in TEST_F()
163 module->entry_computation()->root_instruction(), in TEST_F()
185 auto root = module->entry_computation()->root_instruction(); in TEST_F()
201 auto root = module->entry_computation()->root_instruction(); in TEST_F()
210 auto root = module->entry_computation()->root_instruction(); in TEST_F()
251 auto root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]

123456