Home
last modified time | relevance | path

Searched refs:hlo_string (Results 1 – 25 of 68) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment_test.cc50 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/service/
Dsharding_propagation_test.cc140 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 …]
Dwhile_loop_constant_sinking_test.cc31 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 …]
Dconditional_code_motion_test.cc42 absl::string_view hlo_string = in TEST_F() local
73 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
82 absl::string_view hlo_string = in TEST_F() local
122 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
128 absl::string_view hlo_string = in TEST_F() local
158 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
167 absl::string_view hlo_string = in TEST_F() local
199 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
208 absl::string_view hlo_string = in TEST_F() local
240 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
[all …]
Dwhile_loop_simplifier_test.cc86 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 …]
Dhlo_element_type_converter_test.cc34 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 …]
Dmemory_space_propagation_test.cc39 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 …]
Dhlo_verifier_test.cc158 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 …]
Dstable_sort_expander_test.cc84 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 …]
Dconvolution_4d_expander_test.cc34 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 …]
Dsort_simplifier_test.cc34 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()
Droot_instruction_sinker_test.cc30 absl::string_view hlo_string = R"( in TEST_F() local
56 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
71 absl::string_view hlo_string = R"( in TEST_F() local
100 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
115 absl::string_view hlo_string = R"( in TEST_F() local
127 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
142 absl::string_view hlo_string = R"( in TEST_F() local
157 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
Dhlo_domain_test.cc125 const char* const hlo_string = R"( in TEST_F() local
140 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
168 const char* const hlo_string = R"( in TEST_F() local
183 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
192 const char* const hlo_string = R"( in TEST_F() local
211 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
234 const char* const hlo_string = R"( in TEST_F() local
249 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
258 const char* const hlo_string = R"( in TEST_F() local
273 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
[all …]
Dconvolution_group_converter_test.cc37 string hlo_string = R"(HloModule Convolve1D1Window_0_module in TEST_F() local
46 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
69 string hlo_string = R"(HloModule Convolve1D1Window_0_module in TEST_F() local
78 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
97 string hlo_string = R"(HloModule Convolve1D1Window_0_module in TEST_F() local
105 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
124 string hlo_string = R"(HloModule m in TEST_F() local
132 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
Dspace_to_batch_converter_test.cc36 string hlo_string = R"( in TEST_F() local
48 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
68 string hlo_string = R"( in TEST_F() local
79 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
86 string hlo_string = R"( in TEST_F() local
97 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
117 string hlo_string = R"( in TEST_F() local
130 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
Dtranspose_folding_test.cc62 string hlo_string = R"( in TEST_F() local
73 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
83 string hlo_string = R"( in TEST_F() local
95 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
111 string hlo_string = R"( in TEST_F() local
123 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
139 string hlo_string = R"( in TEST_F() local
151 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
192 string hlo_string = R"( in TEST_F() local
209 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
[all …]
Dconditional_simplifier_test.cc176 absl::string_view hlo_string = in TEST_F() local
206 auto status = ParseAndReturnVerifiedModule(hlo_string); in TEST_F()
226 absl::string_view hlo_string = R"( in TEST_F() local
265 auto status = ParseAndReturnVerifiedModule(hlo_string); in TEST_F()
283 absl::string_view hlo_string = in TEST_F() local
312 auto status = ParseAndReturnVerifiedModule(hlo_string); in TEST_F()
326 absl::string_view hlo_string = in TEST_F() local
356 auto status = ParseAndReturnVerifiedModule(hlo_string); in TEST_F()
372 absl::string_view hlo_string = in TEST_F() local
402 auto status = ParseAndReturnVerifiedModule(hlo_string); in TEST_F()
[all …]
Dtopk_rewriter_test.cc64 const std::string hlo_string = R"( in TEST_F() local
79 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
94 const std::string hlo_string = R"( in TEST_F() local
109 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
124 const std::string hlo_string = R"( in TEST_F() local
139 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
Dwhile_util_test.cc37 const char* const hlo_string = R"( in GetParsedModule() local
58 TF_ASSIGN_OR_RETURN(auto module, ParseAndReturnVerifiedModule(hlo_string)); in GetParsedModule()
137 const char* const hlo_string = R"( in TEST_F() local
160 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
175 const char* const hlo_string = R"( in TEST_F() local
201 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_test.cc70 absl::string_view hlo_string = R"( in TEST_F() local
80 auto module_status = PartitionComputation(hlo_string, /*num_devices=*/4); in TEST_F()
88 absl::string_view hlo_string = R"( in TEST_F() local
97 PartitionComputation(hlo_string, /*num_devices=*/2)); in TEST_F()
107 absl::string_view hlo_string = R"( in TEST_F() local
116 PartitionComputation(hlo_string, /*num_devices=*/2)); in TEST_F()
126 absl::string_view hlo_string = R"( in TEST_F() local
136 PartitionComputation(hlo_string, /*num_devices=*/2)); in TEST_F()
152 absl::string_view hlo_string = R"( in TEST_F() local
161 PartitionComputation(hlo_string, /*num_devices=*/2)); in TEST_F()
[all …]
Dcanonicalize_all_gather_for_cse_test.cc54 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()
93 absl::string_view hlo_string = R"( in TEST_F() local
103 auto module_status = RunPass(hlo_string); in TEST_F()
113 absl::string_view hlo_string = R"( in TEST_F() local
123 auto module_status = RunPass(hlo_string); in TEST_F()
133 absl::string_view hlo_string = R"( in TEST_F() local
143 auto module_status = RunPass(hlo_string); in TEST_F()
[all …]
Dschedule_aware_all_gather_cse_test.cc46 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
99 auto module_status = RunPass(hlo_string); in TEST_F()
109 absl::string_view hlo_string = R"( in TEST_F() local
120 auto module_status = RunPass(hlo_string); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_atomic_test.cc30 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()
122 const char* hlo_string = R"( in TEST_F() local
144 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dsample_text_test.cc38 const string& hlo_string = R"( in TEST_F() local
49 EXPECT_TRUE(RunAndCompareNoHloPasses(hlo_string, ErrorSpec{0.0001})); in TEST_F()
53 const string& hlo_string = R"( in TEST_F() local
62 EXPECT_TRUE(RunAndCompare(hlo_string, nullopt)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor_test.cc29 const string& hlo_string = R"( in TEST_F() local
40 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
65 const string& hlo_string = R"( in TEST_F() local
78 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
110 const string& hlo_string = R"( in TEST_F() local
122 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()

123