Home
last modified time | relevance | path

Searched refs:kModuleStr (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_dimension_simplifier_test.cc53 const char* kModuleStr = R"( in TEST_F() local
63 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
72 const char* kModuleStr = R"( in TEST_F() local
82 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
88 const char* kModuleStr = R"( in TEST_F() local
98 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
106 const char* kModuleStr = R"( in TEST_F() local
116 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
122 const char* kModuleStr = R"( in TEST_F() local
132 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
[all …]
H A Dslice_sinker_test.cc45 const char* kModuleStr = R"( in TEST_F() local
63 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
85 const char* kModuleStr = R"( in TEST_F() local
103 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
128 const char* kModuleStr = R"( in TEST_F() local
149 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
180 const char* kModuleStr = R"( in TEST_F() local
198 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
223 const char* kModuleStr = R"( in TEST_F() local
238 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
[all …]
H A Dscatter_simplifier_test.cc33 constexpr absl::string_view kModuleStr = R"( in TEST_F() local
59 RunAndFilecheckHloRewrite(kModuleStr, ScatterSimplifier(), R"( in TEST_F()
77 constexpr absl::string_view kModuleStr = R"( in TEST_F() local
97 RunAndFilecheckHloRewrite(kModuleStr, ScatterSimplifier(), R"( in TEST_F()
109 constexpr absl::string_view kModuleStr = R"( in TEST_F() local
129 RunAndFilecheckHloRewrite(kModuleStr, ScatterSimplifier(), std::nullopt); in TEST_F()
134 constexpr absl::string_view kModuleStr = R"( in TEST_F() local
154 RunAndFilecheckHloRewrite(kModuleStr, ScatterSimplifier(), R"( in TEST_F()
165 constexpr absl::string_view kModuleStr = R"( in TEST_F() local
185 RunAndFilecheckHloRewrite(kModuleStr, ScatterSimplifier(), R"( in TEST_F()
[all …]
H A Dgather_simplifier_test.cc32 constexpr absl::string_view kModuleStr = R"( in TEST_F() local
46 RunAndFilecheckHloRewrite(kModuleStr, GatherSimplifier(), R"( in TEST_F()
63 constexpr absl::string_view kModuleStr = R"( in TEST_F() local
77 RunAndFilecheckHloRewrite(kModuleStr, GatherSimplifier(), R"( in TEST_F()
87 constexpr absl::string_view kModuleStr = R"( in TEST_F() local
101 RunAndFilecheckHloRewrite(kModuleStr, GatherSimplifier(), R"( in TEST_F()
112 constexpr absl::string_view kModuleStr = R"( in TEST_F() local
126 RunAndFilecheckHloRewrite(kModuleStr, GatherSimplifier(), R"( in TEST_F()
137 constexpr absl::string_view kModuleStr = R"( in TEST_F() local
153 RunAndFilecheckHloRewrite(kModuleStr, GatherSimplifier(), R"( in TEST_F()
[all …]
H A Dhlo_verifier_test.cc1521 const char* const kModuleStr = R"( in TEST_F() local
1535 ParseAndReturnUnverifiedModule(kModuleStr)); in TEST_F()
1542 const char* const kModuleStr = R"( in TEST_F() local
1556 ParseAndReturnUnverifiedModule(kModuleStr)); in TEST_F()
1565 const char* const kModuleStr = R"( in TEST_F() local
1580 ParseAndReturnUnverifiedModule(kModuleStr)); in TEST_F()
1590 const char* const kModuleStr = R"( in TEST_F() local
1600 ParseAndReturnUnverifiedModule(kModuleStr)); in TEST_F()
1676 const char* const kModuleStr = R"( in TEST_F() local
1688 ParseAndReturnUnverifiedModule(kModuleStr, config)); in TEST_F()
[all …]
H A Dscatter_expander_test.cc46 const char* kModuleStr = R"( in TEST_F() local
65 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
76 const char* kModuleStr = R"( in TEST_F() local
100 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
112 const char* kModuleStr = R"( in TEST_F() local
133 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
145 const char* kModuleStr = R"( in TEST_F() local
171 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
183 const char* kModuleStr = R"( in TEST_F() local
202 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
[all …]
H A Dpattern_matcher_test.cc34 constexpr char kModuleStr[] = R"(HloModule two_plus_two_module in TEST_F() local
41 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
153 constexpr char kModuleStr[] = R"( in TEST_F() local
165 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
177 constexpr char kModuleStr[] = R"( in TEST_F() local
185 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
196 constexpr char kModuleStr[] = R"( in TEST_F() local
199 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
219 constexpr char kModuleStr[] = R"( in TEST_F() local
230 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
[all …]
H A Dwhile_loop_trip_count_annotator_test.cc31 const char* kModuleStr = R"( in TEST_F() local
54 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
67 const char* kModuleStr = R"( in TEST_F() local
90 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
103 const char* kModuleStr = R"( in TEST_F() local
126 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
139 const char* kModuleStr = R"( in TEST_F() local
162 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
178 const char* kModuleStr = R"( in TEST_F() local
201 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
H A Dalgebraic_simplifier_test.cc80 const char* kModuleStr = R"( in TEST_F() local
91 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
102 const char* kModuleStr = R"( in TEST_F() local
113 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
123 const char* kModuleStr = R"( in TEST_F() local
131 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
139 const char* kModuleStr = R"( in TEST_F() local
151 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
162 const char* kModuleStr = R"( in TEST_F() local
172 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
[all …]
H A Dreal_imag_expander_test.cc42 const char* kModuleStr = R"( in TEST_F() local
50 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
61 const char* kModuleStr = R"( in TEST_F() local
69 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
82 const char* kModuleStr = R"( in TEST_F() local
92 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
102 const char* kModuleStr = R"( in TEST_F() local
111 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
H A Dall_reduce_simplifier_test.cc42 const char* kModuleStr = R"( in TEST_F() local
81 kModuleStr, /*replica_count=*/8)); in TEST_F()
95 const char* kModuleStr = R"( in TEST_F() local
117 kModuleStr, /*replica_count=*/8)); in TEST_F()
127 const char* kModuleStr = R"( in TEST_F() local
158 kModuleStr, /*replica_count=*/8)); in TEST_F()
170 const char* kModuleStr = R"( in TEST_F() local
188 kModuleStr, /*replica_count=*/8)); in TEST_F()
H A Dsimplify_fp_conversions_test.cc35 const absl::string_view kModuleStr = R"( in TEST_F() local
45 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
52 const absl::string_view kModuleStr = R"( in TEST_F() local
63 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
72 const absl::string_view kModuleStr = R"( in TEST_F() local
84 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
H A Dchange_op_data_type_test.cc41 const char* const kModuleStr = R"( in TEST_F() local
47 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
62 const char* const kModuleStr = R"( in TEST_F() local
68 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
77 const char* const kModuleStr = R"( in TEST_F() local
87 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
H A Dselect_and_scatter_expander_test.cc27 constexpr absl::string_view kModuleStr = variable
62 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
64 RunAndFilecheckHloRewrite(kModuleStr, SelectAndScatterExpander(), R"( in TEST_F()
71 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
73 RunAndFilecheckHloRewrite(kModuleStr, SelectAndScatterExpander(), R"( in TEST_F()
H A Dhlo_constant_folding_test.cc275 const char* const kModuleStr = R"( in TEST_F() local
287 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
296 const char* const kModuleStr = R"( in TEST_F() local
313 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
321 const char* const kModuleStr = R"( in TEST_F() local
335 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
350 constexpr absl::string_view kModuleStr = R"( in TEST_F() local
366 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
H A Ddynamic_update_slice_test.cc34 const char kModuleStr[] = R"( in XLA_TEST_F() local
69 ParseAndReturnVerifiedModule(kModuleStr)); in XLA_TEST_F()
87 const char kModuleStr[] = R"( in XLA_TEST_F() local
192 RunAndCompare(ParseAndReturnVerifiedModule(kModuleStr).ValueOrDie(), in XLA_TEST_F()
H A Dlogistic_expander_test.cc54 const char* kModuleStr = R"( in TEST_F() local
61 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
81 const char* kModuleStr = R"( in TEST_F() local
88 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
H A Dtuple_simplifier_test.cc290 const char* kModuleStr = R"( in TEST_F() local
299 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
H A Dhlo_cse_test.cc918 const char* kModuleStr = R"( in TEST_P() local
930 absl::Substitute(kModuleStr, op))); in TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dcollective_ops_test.cc485 const char* const kModuleStr = R"( in XLA_TEST_F() local
503 ParseAndReturnVerifiedModule(kModuleStr, config)); in XLA_TEST_F()
516 const absl::string_view kModuleStr = R"( in XLA_TEST_F() local
535 ParseAndReturnVerifiedModule(kModuleStr, config)); in XLA_TEST_F()
548 const absl::string_view kModuleStr = R"( in XLA_TEST_F() local
568 ParseAndReturnVerifiedModule(kModuleStr, config)); in XLA_TEST_F()
584 const char* const kModuleStr = R"( in XLA_TEST_F() local
596 ParseAndReturnVerifiedModule(kModuleStr)); in XLA_TEST_F()
609 const char* const kModuleStr = R"( in XLA_TEST_F() local
625 ParseAndReturnVerifiedModule(kModuleStr, config)); in XLA_TEST_F()
[all …]
H A Dcustom_call_test.cc216 const char* kModuleStr = R"( in XLA_TEST_F() local
225 ParseAndReturnVerifiedModule(kModuleStr)); in XLA_TEST_F()
293 const char* const kModuleStr = R"( in XLA_TEST_F() local
308 ParseAndReturnVerifiedModule(kModuleStr)); in XLA_TEST_F()
316 const char* const kModuleStr = R"( in XLA_TEST_F() local
327 ParseAndReturnVerifiedModule(kModuleStr)); in XLA_TEST_F()
H A Dget_dimension_size_test.cc28 const char* const kModuleStr = R"( in TEST_F() local
43 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
H A Dptxas_bug_120501638.cc30 const char* const kModuleStr = R"( in TEST_F() local
77 ParseAndReturnVerifiedModule(kModuleStr, config)); in TEST_F()