Home
last modified time | relevance | path

Searched refs:entry_computation (Results 1 – 25 of 215) sorted by relevance

123456789

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_creation_utils_test.cc35 HloComputation** entry_computation) { in CreateModuleWithProgramShape() argument
40 *entry_computation = module->AddEntryComputation( in CreateModuleWithProgramShape()
43 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape()
50 HloComputation** entry_computation, PrimitiveType primitive_type_output) { in CreateModuleWithProgramShape() argument
55 *entry_computation = module->AddEntryComputation( in CreateModuleWithProgramShape()
58 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape()
65 HloComputation* entry_computation; in TEST_F() local
69 &entry_computation); in TEST_F()
73 entry_computation->set_root_instruction(first_1_dims_collapsed); in TEST_F()
84 HloComputation* entry_computation; in TEST_F() local
[all …]
H A Dar_crs_combiner_test.cc44 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F()
48 i1, module->entry_computation()->parameter_instruction(0))); 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()
[all …]
H A Dhlo_module_dce_test.cc112 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
114 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
152 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
154 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
190 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
193 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
196 auto while_loops = GetWhileLoops(module->entry_computation()); in TEST_F()
234 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
237 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
240 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
[all …]
H A Dwhile_loop_all_reduce_code_motion_test.cc95 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F()
96 module->entry_computation()->instructions().end(), in TEST_F()
108 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F()
109 module->entry_computation()->instructions().end(), in TEST_F()
194 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F()
195 module->entry_computation()->instructions().end(), in TEST_F()
204 absl::c_copy_if(module->entry_computation()->instructions(), in TEST_F()
273 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F()
274 module->entry_computation()->instructions().end(), in TEST_F()
282 HloInstruction* new_root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
H A Dcall_graph_test.cc119 HloComputation* entry_computation = in TEST_F() local
127 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation); in TEST_F()
129 EXPECT_EQ(entry_computation, entry_node.computation()); in TEST_F()
145 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local
151 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation); in TEST_F()
152 EXPECT_EQ(entry_computation, entry_node.computation()); in TEST_F()
158 EXPECT_TRUE(call_graph->GetComputationCallers(entry_computation).empty()); in TEST_F()
178 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local
188 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation); in TEST_F()
189 EXPECT_EQ(entry_computation, entry_node.computation()); in TEST_F()
[all …]
H A Dhlo_rematerialization_test.cc160 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 …]
H A 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 …]
H A Dwhile_util_test.cc35 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, &param0, &param1, &param2)); 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 …]
H A Dfusion_node_indexing_evaluation_test.cc89 HloInstruction* sub = module->entry_computation()->root_instruction(); in TEST_F()
91 InstructionFusionForTesting().Fuse(add, sub, module->entry_computation()); in TEST_F()
104 HloInstruction* sub = module->entry_computation()->root_instruction(); in TEST_F()
109 instruction_fusion.Fuse(slice1, sub, module->entry_computation()); in TEST_F()
111 instruction_fusion.Fuse(slice2, fusion, module->entry_computation()); in TEST_F()
145 HloInstruction* add2 = module->entry_computation()->root_instruction(); in TEST_F()
150 instruction_fusion.Fuse(slice2_0, add2, module->entry_computation()); in TEST_F()
155 instruction_fusion.Fuse(slice2_1, fusion, module->entry_computation()); in TEST_F()
160 instruction_fusion.Fuse(add1, fusion, module->entry_computation()); in TEST_F()
166 instruction_fusion.Fuse(slice1_0, fusion, module->entry_computation()); in TEST_F()
[all …]
H A Dinstruction_fusion_test.cc57 HloInstruction* sub = module->entry_computation()->root_instruction(); in TEST_F()
60 InstructionFusionForTesting().Fuse(add, sub, module->entry_computation()); in TEST_F()
81 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
84 abs, root, module->entry_computation()); in TEST_F()
101 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
105 abs, tanh, module->entry_computation()); in TEST_F()
196 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
302 root = module->entry_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 …]
H A Dtranspose_folding_test.cc63 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
105 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
160 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
182 HloComputation* entry_computation = in TEST_F() local
185 {add, sub, mul}, "entry", entry_computation); in TEST_F()
186 EXPECT_EQ(call, entry_computation->root_instruction()); in TEST_F()
260 HloComputation* entry_computation = in TEST_F() local
267 entry_computation->instructions().begin(), in TEST_F()
268 entry_computation->instructions().end()); in TEST_F()
318 HloComputation* entry_computation = in TEST_F() local
[all …]
H A 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()
231 LOG(INFO) << module->entry_computation()->ToString(); in TEST_F()
233 module->entry_computation()->root_instruction(), in TEST_F()
238 const HloInstruction* cc = module->entry_computation() in TEST_F()
262 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
[all …]
H A Dwhile_loop_simplifier_test.cc39 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 …]
H A 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()
370 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
438 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
516 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
722 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
809 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
854 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A 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 …]
H A Dgpu_conv_rewriter_test.cc126 HloComputation* entry_computation = in TEST_F() local
129 ASSERT_THAT(entry_computation->root_instruction(), in TEST_F()
135 entry_computation->root_instruction()->operand(0)->metadata(); in TEST_F()
163 HloComputation* entry_computation = in TEST_F() local
167 entry_computation->root_instruction(), in TEST_F()
193 HloComputation* entry_computation = in TEST_F() local
196 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
223 HloComputation* entry_computation = in TEST_F() local
226 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
252 HloComputation* entry_computation = in TEST_F() local
[all …]
H A Dgpu_layout_assignment_test.cc110 module->entry_computation()->ComputeProgramShape(), in TEST_F()
115 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
136 module->entry_computation()->ComputeProgramShape(), in TEST_F()
142 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
163 module->entry_computation()->ComputeProgramShape(), in TEST_F()
169 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
189 module->entry_computation()->ComputeProgramShape(), in TEST_F()
195 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
216 module->entry_computation()->ComputeProgramShape(), in TEST_F()
222 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
[all …]
H A 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 …]
H A Dcudnn_fused_conv_rewriter_test.cc454 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
491 m->entry_computation()->root_instruction(), in TEST_F()
531 m->entry_computation()->root_instruction(), in TEST_F()
581 m->entry_computation()->root_instruction(), in TEST_F()
629 m->entry_computation()->root_instruction(), in TEST_F()
674 m->entry_computation()->root_instruction(), in TEST_F()
712 m->entry_computation()->root_instruction(), in TEST_F()
751 m->entry_computation()->root_instruction(), in TEST_F()
793 m->entry_computation()->root_instruction(), in TEST_F()
831 m->entry_computation()->root_instruction(), in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/
H A Dhlo_control_flow_flattening_test.cc70 auto root = module->entry_computation()->root_instruction(); in TEST_F()
71 auto while_op = module->entry_computation()->GetInstructionWithName("while"); in TEST_F()
131 auto root = module->entry_computation()->root_instruction(); in TEST_F()
132 auto while_op = module->entry_computation()->GetInstructionWithName("while"); in TEST_F()
185 auto root = module->entry_computation()->root_instruction(); in TEST_F()
186 auto while_op = module->entry_computation()->GetInstructionWithName("while"); in TEST_F()
241 auto fusion = module->entry_computation()->root_instruction(); in TEST_F()
242 auto while_op = module->entry_computation()->GetInstructionWithName("while"); in TEST_F()
264 auto tuple = module->entry_computation()->root_instruction(); in TEST_F()
278 Shape root_shape = module->entry_computation()->root_instruction()->shape(); in TEST_F()
[all …]
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder_test.cc77 auto root = module->entry_computation()->root_instruction(); in TEST_F()
88 auto root = module->entry_computation()->root_instruction(); in TEST_F()
102 auto root = module->entry_computation()->root_instruction(); in TEST_F()
133 auto root = module->entry_computation()->root_instruction(); in TEST_F()
150 module->entry_computation()->root_instruction(), in TEST_F()
166 module->entry_computation()->root_instruction(), in TEST_F()
188 auto root = module->entry_computation()->root_instruction(); in TEST_F()
204 auto root = module->entry_computation()->root_instruction(); in TEST_F()
213 auto root = module->entry_computation()->root_instruction(); in TEST_F()
265 auto root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_test.cc110 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
133 EXPECT_NE(custom_call->parent(), module->entry_computation()); 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()
[all …]
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dgpu_reduce_scatter_creator_test.cc58 return absl::c_count_if(module->entry_computation()->instructions(), in AllReduceCount()
95 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()
[all …]

123456789