| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_reduce_reassociate_test.cc | 51 absl::string_view hlo_string = R"( in TEST_F() local
69 RunPass(hlo_string, /*expect_change=*/true)); in TEST_F()
76 absl::string_view hlo_string = R"( in TEST_F() local
94 RunPass(hlo_string, /*expect_change=*/true)); in TEST_F()
103 absl::string_view hlo_string = R"( in TEST_F() local
127 RunPass(hlo_string, /*expect_change=*/true)); in TEST_F()
138 absl::string_view hlo_string = R"( in TEST_F() local
162 RunPass(hlo_string, /*expect_change=*/true)); in TEST_F()
170 absl::string_view hlo_string = R"( in TEST_F() local
194 RunPass(hlo_string, /*expect_change=*/false)); in TEST_F()
[all …]
|
| D | sharding_propagation_test.cc | 140 const char* const hlo_string = R"( in TEST_P() local
150 ParseAndReturnVerifiedModule(hlo_string)); in TEST_P()
172 const char* const hlo_string = R"( in TEST_F() local
182 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
196 const char* const hlo_string = R"( in TEST_F() local
205 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
219 const char* const hlo_string = R"( in TEST_F() local
228 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
241 const char* const hlo_string = R"( in TEST_P() local
251 ParseAndReturnVerifiedModule(hlo_string)); in TEST_P()
[all …]
|
| D | while_loop_constant_sinking_test.cc | 31 const char* const hlo_string = R"( in TEST_F() local
57 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
69 const char* const hlo_string = R"( in TEST_F() local
97 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
111 const char* const hlo_string = R"( in TEST_F() local
138 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
157 const char* const hlo_string = R"( in TEST_F() local
186 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
200 const char* const hlo_string = R"( in TEST_F() local
228 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
[all …]
|
| D | all_gather_combiner_test.cc | 63 const char* const hlo_string = R"( in TEST_F() local
75 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
92 const char* const hlo_string = R"( in TEST_F() local
111 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
132 const char* const hlo_string = R"( in TEST_F() local
144 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
158 const char* const hlo_string = R"( in TEST_F() local
170 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
183 const char* const hlo_string = R"( in TEST_F() local
193 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
[all …]
|
| D | while_loop_simplifier_test.cc | 86 string hlo_string = absl::StrReplaceAll( in MakeModuleWithSimpleLoop() local
88 return ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in MakeModuleWithSimpleLoop()
124 string hlo_string = absl::StrReplaceAll( in MakeModuleWithSimpleLoopTupleElementLoopBound() local
126 return ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in MakeModuleWithSimpleLoopTupleElementLoopBound()
243 const string hlo_string = R"( in TEST_F() local
263 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
271 const string hlo_string = R"( in TEST_F() local
295 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
304 const string hlo_string = R"( in TEST_F() local
324 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
[all …]
|
| D | conditional_code_motion_test.cc | 43 absl::string_view hlo_string = in TEST_F() local
74 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
83 absl::string_view hlo_string = in TEST_F() local
123 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
129 absl::string_view hlo_string = in TEST_F() local
159 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
168 absl::string_view hlo_string = in TEST_F() local
200 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
209 absl::string_view hlo_string = in TEST_F() local
241 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
[all …]
|
| D | memory_space_propagation_test.cc | 39 absl::string_view hlo_string = R"( in TEST_F() local
65 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
68 TF_ASSERT_OK_AND_ASSIGN(auto ref, ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
73 absl::string_view hlo_string = R"( in TEST_F() local
124 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
134 absl::string_view hlo_string = R"( in TEST_F() local
193 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
204 absl::string_view hlo_string = R"( in TEST_F() local
267 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
278 absl::string_view hlo_string = R"( in TEST_F() local
[all …]
|
| D | all_gather_broadcast_reorder_test.cc | 53 absl::string_view hlo_string = R"( in TEST_F() local
62 RunPass(hlo_string, PassOutput::NonUniformAGPattern); in TEST_F()
66 absl::string_view hlo_string = R"( in TEST_F() local
75 RunPass(hlo_string, PassOutput::UniformAGPattern); in TEST_F()
79 absl::string_view hlo_string = R"( in TEST_F() local
88 RunPass(hlo_string, PassOutput::UniformAGPattern); in TEST_F()
92 absl::string_view hlo_string = R"( in TEST_F() local
102 RunPass(hlo_string, PassOutput::UniformAGPattern); in TEST_F()
106 absl::string_view hlo_string = R"( in TEST_F() local
115 RunPass(hlo_string, PassOutput::NoChange); in TEST_F()
|
| D | hlo_element_type_converter_test.cc | 34 const string& hlo_string = R"( in TEST_F() local
42 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
49 const string& hlo_string = R"( in TEST_F() local
58 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
65 const string& hlo_string = R"( in TEST_F() local
77 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
87 const string& hlo_string = R"( in TEST_F() local
105 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
120 const string& hlo_string = R"( in TEST_F() local
129 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
[all …]
|
| D | hlo_verifier_test.cc | 158 const char* const hlo_string = R"( in TEST_F() local
171 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
180 const char* const hlo_string = R"( in TEST_F() local
201 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
210 const char* const hlo_string = R"( in TEST_F() local
236 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
256 const char* const hlo_string = R"( in TEST_F() local
267 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
275 const char* const hlo_string = R"( in TEST_F() local
286 ParseAndReturnUnverifiedModule(hlo_string)); in TEST_F()
[all …]
|
| D | reduce_scatter_decomposer_test.cc | 101 absl::string_view hlo_string = R"( in TEST_F() local
115 RunPass(hlo_string, PassAction::kTrivialGroups, in TEST_F()
121 absl::string_view hlo_string = R"( in TEST_F() local
135 RunPass(hlo_string, PassAction::kTableLookup, in TEST_F()
141 absl::string_view hlo_string = R"( in TEST_F() local
157 RunPass(hlo_string, PassAction::kTrivialGroups, in TEST_F()
164 absl::string_view hlo_string = R"( in TEST_F() local
182 RunPass(hlo_string, PassAction::kTrivialGroups, in TEST_F()
188 absl::string_view hlo_string = R"( in TEST_F() local
202 RunPass(hlo_string, PassAction::kTableLookup, in TEST_F()
[all …]
|
| D | stable_sort_expander_test.cc | 84 const char* hlo_string = R"( in TEST_F() local
103 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
116 const char* hlo_string = R"( in TEST_F() local
149 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
161 const char* hlo_string = R"( in TEST_F() local
179 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
196 const char* hlo_string = R"( in TEST_F() local
215 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
223 const char* hlo_string = R"( in TEST_F() local
242 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
[all …]
|
| D | space_to_batch_converter_test.cc | 36 string hlo_string = R"( in TEST_F() local
48 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
69 string hlo_string = R"( in TEST_F() local
93 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
103 string hlo_string = R"( in TEST_F() local
114 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
122 string hlo_string = R"( in TEST_F() local
137 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
145 string hlo_string = R"( in TEST_F() local
156 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
[all …]
|
| D | convolution_4d_expander_test.cc | 34 string hlo_string = R"(HloModule convolution_4d_fp32 in TEST_F() local
42 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
59 string hlo_string = R"(HloModule convolution_4d_fp32 in TEST_F() local
67 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
86 string hlo_string = R"(HloModule convolution_4d_fp32 in TEST_F() local
94 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
111 string hlo_string = R"(HloModule convolution_4d_fp32 in TEST_F() local
119 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
130 string hlo_string = R"(HloModule convolution_4d_fp32 in TEST_F() local
138 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
[all …]
|
| D | sharding_remover_test.cc | 31 const char* const hlo_string = R"( in TEST_F() local
41 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
49 const char* const hlo_string = R"( in TEST_F() local
59 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
67 const char* const hlo_string = R"( in TEST_F() local
77 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
85 const char* const hlo_string = R"( in TEST_F() local
104 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
|
| D | reduce_scatter_combiner_test.cc | 57 absl::string_view hlo_string = R"( in TEST_F() local
75 RunPass(hlo_string, /*expect_change=*/true)); in TEST_F()
80 absl::string_view hlo_string = R"( in TEST_F() local
100 RunPass(hlo_string, /*expect_change=*/true)); in TEST_F()
106 absl::string_view hlo_string = R"( in TEST_F() local
123 RunPass(hlo_string, /*expect_change=*/false)); in TEST_F()
127 absl::string_view hlo_string = R"( in TEST_F() local
145 RunPass(hlo_string, /*expect_change=*/false)); in TEST_F()
|
| D | sort_simplifier_test.cc | 34 const char* hlo_string = R"( in TEST_F() local
53 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
66 const char* hlo_string = R"( in TEST_F() local
91 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
104 const char* hlo_string = R"( in TEST_F() local
122 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
129 const char* hlo_string = R"( in TEST_F() local
148 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
|
| D | collectives_schedule_linearizer_test.cc | 56 absl::string_view hlo_string = R"( in TEST_F() local
75 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
92 absl::string_view hlo_string = R"( in TEST_F() local
111 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
117 absl::string_view hlo_string = R"( in TEST_F() local
136 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
142 absl::string_view hlo_string = R"( in TEST_F() local
162 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_reduce_scatter_creator_test.cc | 66 absl::string_view hlo_string = R"( in TEST_F() local
91 TF_ASSERT_OK_AND_ASSIGN(auto module, RunPass(hlo_string, in TEST_F()
104 absl::string_view hlo_string = R"( in TEST_F() local
130 TF_ASSERT_OK_AND_ASSIGN(auto module, RunPass(hlo_string, in TEST_F()
140 absl::string_view hlo_string = R"( in TEST_F() local
163 TF_ASSERT_OK_AND_ASSIGN(auto module, RunPass(hlo_string, in TEST_F()
173 absl::string_view hlo_string = R"( in TEST_F() local
197 TF_ASSERT_OK_AND_ASSIGN(auto module, RunPass(hlo_string, in TEST_F()
210 absl::string_view hlo_string = R"( in TEST_F() local
234 TF_ASSERT_OK_AND_ASSIGN(auto module, RunPass(hlo_string, in TEST_F()
[all …]
|
| D | gpu_atomic_test.cc | 30 const char* hlo_string = R"( in TEST_F() local
51 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
57 const char* hlo_string = R"( in TEST_F() local
78 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
84 const char* hlo_string = R"( in TEST_F() local
106 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
121 const char* hlo_string = R"( in TEST_F() local
143 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 71 absl::string_view hlo_string = R"( in TEST_F() local
81 auto module_status = PartitionComputation(hlo_string, /*num_devices=*/4); in TEST_F()
89 absl::string_view hlo_string = R"( in TEST_F() local
98 PartitionComputation(hlo_string, /*num_devices=*/2)); in TEST_F()
108 absl::string_view hlo_string = R"( in TEST_F() local
117 PartitionComputation(hlo_string, /*num_devices=*/2)); in TEST_F()
127 absl::string_view hlo_string = R"( in TEST_F() local
137 PartitionComputation(hlo_string, /*num_devices=*/2)); in TEST_F()
153 absl::string_view hlo_string = R"( in TEST_F() local
162 PartitionComputation(hlo_string, /*num_devices=*/2)); in TEST_F()
[all …]
|
| D | schedule_aware_collective_ops_cse_test.cc | 46 absl::string_view hlo_string = R"( in TEST_F() local
57 auto module_status = RunPass(hlo_string); in TEST_F()
67 absl::string_view hlo_string = R"( in TEST_F() local
78 auto module_status = RunPass(hlo_string); in TEST_F()
88 absl::string_view hlo_string = R"( in TEST_F() local
101 auto module_status = RunPass(hlo_string); in TEST_F()
111 absl::string_view hlo_string = R"( in TEST_F() local
124 auto module_status = RunPass(hlo_string); in TEST_F()
134 absl::string_view hlo_string = R"( in TEST_F() local
147 auto module_status = RunPass(hlo_string); in TEST_F()
[all …]
|
| D | canonicalize_all_gather_for_cse_test.cc | 54 absl::string_view hlo_string = R"( in TEST_F() local
63 auto module_status = RunPass(hlo_string); in TEST_F()
73 absl::string_view hlo_string = R"( in TEST_F() local
83 auto module_status = RunPass(hlo_string); in TEST_F()
94 absl::string_view hlo_string = R"( in TEST_F() local
104 auto module_status = RunPass(hlo_string); in TEST_F()
115 absl::string_view hlo_string = R"( in TEST_F() local
125 auto module_status = RunPass(hlo_string); in TEST_F()
135 absl::string_view hlo_string = R"( in TEST_F() local
145 auto module_status = RunPass(hlo_string); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment_test.cc | 50 const string hlo_string = R"( in TEST_F() local
61 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
68 const string hlo_string = R"( in TEST_F() local
89 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
95 const string hlo_string = R"( in TEST_F() local
106 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
112 const string hlo_string = R"( in TEST_F() local
123 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
133 const string hlo_string = R"( in TEST_F() local
168 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_control_flow_flattening_test.cc | 30 absl::string_view hlo_string = R"( in TEST_F() local
55 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
84 absl::string_view hlo_string = R"( in TEST_F() local
109 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
132 absl::string_view hlo_string = R"( in TEST_F() local
165 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
181 absl::string_view hlo_string = R"( in TEST_F() local
189 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
203 absl::string_view hlo_string = R"( in TEST_F() local
211 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
[all …]
|