Home
last modified time | relevance | path

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

12345678

/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()
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 …]
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()
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/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 …]
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()
Dalgebraic_simplifier_test.cc70 HloInstruction* root = computation->root_instruction(); in TEST_F()
74 root = computation->root_instruction(); in TEST_F()
94 m->entry_computation()->root_instruction(), in TEST_F()
114 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
132 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
153 m->entry_computation()->root_instruction(), in TEST_F()
173 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
196 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
218 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
276 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.cc55 EXPECT_THAT(computation->root_instruction(), in TEST_F()
62 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F()
63 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<int64>(), in TEST_F()
77 EXPECT_THAT(computation->root_instruction(), in TEST_F()
84 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F()
85 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<float>(), in TEST_F()
99 EXPECT_THAT(computation->root_instruction(), in TEST_F()
106 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F()
107 EXPECT_EQ(computation->root_instruction()->literal().Get<int64>({0}), 42); in TEST_F()
108 EXPECT_EQ(computation->root_instruction()->literal().Get<int64>({1}), 19); in TEST_F()
[all …]
Dhlo_cost_analysis_test.cc148 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 …]
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 …]
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()
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()
182 EXPECT_THAT(computation->root_instruction(), op::Constant()); in TEST_F()
200 outfeed_computation->root_instruction()->shape(), /*operands=*/{}, in TEST_F()
Dlogistic_expander_test.cc63 HloInstruction* root = computation->root_instruction(); in TEST_F()
67 root = computation->root_instruction(); in TEST_F()
68 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
90 HloInstruction* root = computation->root_instruction(); in TEST_F()
94 root = computation->root_instruction(); in TEST_F()
95 EXPECT_THAT(m->entry_computation()->root_instruction(), 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()
Dwhile_loop_constant_sinking_test.cc64 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
104 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
145 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
193 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
235 EXPECT_THAT(while_body->root_instruction(), in TEST_F()
281 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F()
321 EXPECT_THAT(while_condition->root_instruction(), in TEST_F()
364 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F()
414 EXPECT_THAT(while_condition->root_instruction(), in TEST_F()
Dconditional_to_select_test.cc72 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
75 EXPECT_THAT(mapped->root_instruction(), in TEST_F()
125 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
129 mapped->root_instruction(), in TEST_F()
181 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
184 EXPECT_THAT(mapped->root_instruction(), in TEST_F()
Dconvolution_group_converter_test.cc49 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()
87 root = computation->root_instruction(); in TEST_F()
108 HloInstruction* root = computation->root_instruction(); in TEST_F()
114 root = computation->root_instruction(); in TEST_F()
135 HloInstruction* root = computation->root_instruction(); in TEST_F()
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 …]
Dwhile_util_test.cc77 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST_F()
87 entry_computation->root_instruction(), in TEST_F()
95 EXPECT_THAT(new_while_instr->while_body()->root_instruction(), in TEST_F()
108 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST_F()
121 entry_computation->root_instruction(), in TEST_F()
129 EXPECT_THAT(new_while_instr->while_body()->root_instruction(), in TEST_F()
204 HloInstruction* while_instr = main->root_instruction(); in TEST_F()
Dtopk_rewriter_test.cc85 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()
146 module->entry_computation()->root_instruction(), in TEST_F()
152 ->root_instruction() in TEST_F()
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()
/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/service/spmd/
Dspmd_partitioner_test.cc99 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 …]
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc76 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 …]

12345678