Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils_test.cc33 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; in TEST_F() local
52 &entry_computation); in TEST_F()
56 entry_computation->set_root_instruction(first_1_dims_collapsed); in TEST_F()
67 HloComputation* entry_computation; in TEST_F() local
71 &entry_computation); in TEST_F()
75 entry_computation->set_root_instruction(first_2_dims_collapsed); in TEST_F()
90 HloComputation* entry_computation; in TEST_F() local
[all …]
Dhlo_module_dce_test.cc102 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
104 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
142 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
144 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
180 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
183 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
186 EXPECT_TRUE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
223 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
226 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
229 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F()
[all …]
Dar_crs_combiner_test.cc43 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 …]
Dwhile_util_test.cc30 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule() argument
56 *entry_computation = module->entry_computation(); in GetParsedModule()
57 *param0 = (*entry_computation)->parameter_instruction(0); in GetParsedModule()
58 *param1 = (*entry_computation)->parameter_instruction(1); in GetParsedModule()
59 *param2 = (*entry_computation)->parameter_instruction(2); in GetParsedModule()
66 HloComputation* entry_computation; in TEST() local
70 GetParsedModule(&entry_computation, &param0, &param1, &param2)); in TEST()
72 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST()
82 entry_computation->root_instruction(), in TEST()
97 HloComputation* entry_computation; in TEST() local
[all …]
Dtuple_util_test.cc28 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule() argument
42 *entry_computation = module->entry_computation(); in GetParsedModule()
43 *param0 = (*entry_computation)->parameter_instruction(0); in GetParsedModule()
44 *param1 = (*entry_computation)->parameter_instruction(1); in GetParsedModule()
51 HloComputation* entry_computation; in TEST() local
55 GetParsedModule(&entry_computation, &param0, &param1)); in TEST()
65 HloComputation* entry_computation; in TEST() local
69 GetParsedModule(&entry_computation, &param0, &param1)); in TEST()
Dcall_graph_test.cc118 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 …]
Dhlo_rematerialization_test.cc240 HloComputation* entry_computation = in TEST_F() local
244 EXPECT_EQ(entry_computation->instruction_count(), 7); in TEST_F()
256 EXPECT_EQ(entry_computation->instruction_count(), 8); in TEST_F()
276 HloComputation* entry_computation = in TEST_F() local
280 EXPECT_EQ(entry_computation->instruction_count(), 7); in TEST_F()
289 EXPECT_EQ(entry_computation->instruction_count(), 9); in TEST_F()
312 HloComputation* entry_computation = in TEST_F() local
316 EXPECT_EQ(entry_computation->instruction_count(), 7); in TEST_F()
328 EXPECT_EQ(entry_computation->instruction_count(), 9); in TEST_F()
370 HloComputation* entry_computation = in TEST_F() local
[all …]
Dhlo_memory_scheduler_test.cc82 module->schedule().sequence(module->entry_computation()).instructions(); in TEST_F()
83 EXPECT_EQ(module->entry_computation()->instruction_count(), sequence.size()); in TEST_F()
130 schedule.sequence(module->entry_computation()).instructions(); in TEST_F()
131 EXPECT_EQ(module->entry_computation()->instruction_count(), sequence.size()); in TEST_F()
187 EXPECT_EQ(module->entry_computation()->instruction_count(), in TEST_F()
188 schedule.sequence(module->entry_computation()).size()); in TEST_F()
237 EXPECT_EQ(module->entry_computation()->instruction_count(), in TEST_F()
238 schedule.sequence(module->entry_computation()).size()); in TEST_F()
291 auto entry_computation = module->entry_computation(); in TEST_F() local
292 EXPECT_EQ(module->entry_computation()->instruction_count(), in TEST_F()
[all …]
Dtranspose_folding_test.cc76 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
154 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
176 HloComputation* entry_computation = in TEST_F() local
179 {add, sub, mul}, "entry", entry_computation); in TEST_F()
180 EXPECT_EQ(call, entry_computation->root_instruction()); in TEST_F()
252 HloComputation* entry_computation = in TEST_F() local
258 entry_computation->instructions().begin(), in TEST_F()
259 entry_computation->instructions().end()); in TEST_F()
308 HloComputation* entry_computation = in TEST_F() local
314 entry_computation->instructions().begin(), in TEST_F()
[all …]
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 …]
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()
218 HloComputation* computation = m->entry_computation(); in TEST_F()
410 const auto& instrs = m->entry_computation()->instructions(); in TEST_F()
[all …]
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()
59 const HloInstruction* slice = module->entry_computation() in TEST_F()
90 ASSERT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
97 const HloInstruction* slice = module->entry_computation() in TEST_F()
128 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
Dstable_sort_expander_test.cc107 auto root = module->entry_computation()->root_instruction(); in TEST_F()
153 auto root = module->entry_computation()->root_instruction(); in TEST_F()
183 auto root = module->entry_computation()->root_instruction(); in TEST_F()
251 auto root = module->entry_computation()->root_instruction(); in TEST_F()
288 auto root = module->entry_computation()->root_instruction(); in TEST_F()
319 auto root = module->entry_computation()->root_instruction(); in TEST_F()
350 auto root = module->entry_computation()->root_instruction(); in TEST_F()
Dhlo_module_group_test.cc53 group.module(0).entry_computation()->instructions(), in TEST_F()
61 group_copy.module(0).entry_computation()->instructions(), in TEST_F()
96 group.module(0).entry_computation()->instructions(), in TEST_F()
98 EXPECT_THAT(group.module(1).entry_computation()->instructions(), in TEST_F()
135 group.module(0).entry_computation()->instructions(), in TEST_F()
137 EXPECT_THAT(group.module(1).entry_computation()->instructions(), in TEST_F()
Dhlo_schedule_test.cc63 schedule.sequence(module->entry_computation()).instructions(); in TEST_F()
71 schedule.sequence(module->entry_computation()).instructions()); in TEST_F()
97 HloComputation* entry = module->entry_computation(); in TEST_F()
148 HloComputation* entry = module->entry_computation(); in TEST_F()
191 HloComputation* entry = module->entry_computation(); in TEST_F()
253 module->entry_computation()->root_instruction()->operand(0); in TEST_F()
322 module->entry_computation()->root_instruction()->mutable_operand(0); in TEST_F()
/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 …]
Dcudnn_conv_rewriter_test.cc124 HloComputation* entry_computation = in TEST_F() local
127 ASSERT_THAT(entry_computation->root_instruction(), in TEST_F()
133 entry_computation->root_instruction()->operand(0)->metadata(); in TEST_F()
160 HloComputation* entry_computation = in TEST_F() local
163 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
190 HloComputation* entry_computation = in TEST_F() local
193 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
220 HloComputation* entry_computation = in TEST_F() local
223 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F()
249 HloComputation* entry_computation = in TEST_F() local
[all …]
Dgpu_hlo_schedule.cc47 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()
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()
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()
Dinstruction_fusion_test.cc169 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
242 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
263 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
285 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
325 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
353 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
380 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F()
409 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
460 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
534 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F()
[all …]
Dvariadic_op_splitter_test.cc58 auto entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
60 EXPECT_TRUE(Match(entry_computation->root_instruction(), in TEST_F()
73 auto entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
75 EXPECT_TRUE(Match(entry_computation->root_instruction(), 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/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