Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_constant_sinking_test.cc32 const char* const hlo_string = R"( in TEST_F() local
58 ParseHloString(hlo_string)); in TEST_F()
70 const char* const hlo_string = R"( in TEST_F() local
98 ParseHloString(hlo_string)); in TEST_F()
112 const char* const hlo_string = R"( in TEST_F() local
139 ParseHloString(hlo_string)); in TEST_F()
158 const char* const hlo_string = R"( in TEST_F() local
187 ParseHloString(hlo_string)); in TEST_F()
201 const char* const hlo_string = R"( in TEST_F() local
229 ParseHloString(hlo_string)); 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()
230 const string hlo_string = R"( in TEST_F() local
250 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
258 const string hlo_string = R"( in TEST_F() local
282 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
291 const string hlo_string = R"( in TEST_F() local
311 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 …]
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 …]
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()
Dhlo_verifier_test.cc157 const char* const hlo_string = R"( in TEST_F() local
169 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
178 const char* const hlo_string = R"( in TEST_F() local
198 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
207 const char* const hlo_string = R"( in TEST_F() local
217 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
225 const char* const hlo_string = R"( in TEST_F() local
235 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
244 const char* const hlo_string = R"( in TEST_F() local
254 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
[all …]
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.cc38 string hlo_string = R"(HloModule Convolve1D1Window_0_module in TEST_F() local
47 ParseHloString(hlo_string)); in TEST_F()
69 string hlo_string = R"(HloModule Convolve1D1Window_0_module in TEST_F() local
78 ParseHloString(hlo_string)); in TEST_F()
99 string hlo_string = R"(HloModule Convolve1D1Window_0_module in TEST_F() local
107 ParseHloString(hlo_string)); in TEST_F()
Dwhile_util_test.cc32 const char* const hlo_string = R"( in GetParsedModule() local
54 ParseHloString(hlo_string)); in GetParsedModule()
132 const char* const hlo_string = R"( in TEST() local
155 ParseHloString(hlo_string)); in TEST()
170 const char* const hlo_string = R"( in TEST() local
196 ParseHloString(hlo_string)); in TEST()
Dtranspose_folding_test.cc61 string hlo_string = R"( in TEST_F() local
72 ParseHloString(hlo_string)); in TEST_F()
82 string hlo_string = R"( in TEST_F() local
94 ParseHloString(hlo_string)); in TEST_F()
110 string hlo_string = R"( in TEST_F() local
122 ParseHloString(hlo_string)); in TEST_F()
138 string hlo_string = R"( in TEST_F() local
150 ParseHloString(hlo_string)); in TEST_F()
191 string hlo_string = R"( in TEST_F() local
208 ParseHloString(hlo_string)); in TEST_F()
Dhlo_cse_test.cc253 const char* const hlo_string = R"( in TEST_F() local
282 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
294 const char* const hlo_string = R"( in TEST_F() local
332 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
343 const char* const hlo_string = R"( in TEST_F() local
374 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
386 const char* const hlo_string = R"( in TEST_F() local
415 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
662 const char* const hlo_string = R"( in TEST_F() local
685 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
[all …]
Ddfs_hlo_visitor_with_default_test.cc66 const string& hlo_string = R"( in TEST_F() local
83 ParseAndReturnVerifiedModule(hlo_string).ConsumeValueOrDie(); in TEST_F()
/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()
Dir_emission_utils_test.cc26 const char* const hlo_string = R"( in TEST() local
38 ParseHloString(hlo_string)); in TEST()
Dcpu_eigen_tensor_alignment_test.cc31 string hlo_string = R"( in TEST_F() local
42 ParseHloString(hlo_string)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter_test.cc31 string GetOptimizedHlo(absl::string_view hlo_string) { in GetOptimizedHlo() argument
34 ->RunHloPasses(ParseHloString(hlo_string, GetModuleConfigForTest()) in GetOptimizedHlo()
42 void TestMatchWithAllTypes(absl::string_view hlo_string) { in TestMatchWithAllTypes() argument
45 absl::StrReplaceAll(hlo_string, {{"TYPE", type}}); in TestMatchWithAllTypes()
56 void TestNotMatchWithAllTypes(absl::string_view hlo_string) { in TestNotMatchWithAllTypes() argument
59 absl::StrReplaceAll(hlo_string, {{"TYPE", type}}); in TestNotMatchWithAllTypes()
/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()
Dreduce_window_test.cc1407 const string hlo_string = R"( in XLA_TEST_F() local
1420 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0.001})); in XLA_TEST_F()
1424 const string hlo_string = R"( in XLA_TEST_F() local
1437 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0.001})); in XLA_TEST_F()
1441 const string hlo_string = R"( in XLA_TEST_F() local
1454 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0.001})); in XLA_TEST_F()
1458 const string hlo_string = R"( in XLA_TEST_F() local
1472 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0.001})); in XLA_TEST_F()
1476 const string hlo_string = R"( in XLA_TEST_F() local
1490 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0.001})); in XLA_TEST_F()
[all …]
Ddot_operation_test.cc1211 absl::string_view hlo_string = in XLA_TEST_F() local
1222 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-3, 1e-3})); in XLA_TEST_F()
1226 absl::string_view hlo_string = in XLA_TEST_F() local
1237 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-3, 1e-3})); in XLA_TEST_F()
1241 absl::string_view hlo_string = in XLA_TEST_F() local
1252 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-3, 1e-3})); in XLA_TEST_F()
1256 absl::string_view hlo_string = in XLA_TEST_F() local
1267 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{4e-3, 4e-3})); in XLA_TEST_F()
1272 absl::string_view hlo_string = in XLA_TEST_F() local
1288 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{4e-3, 4e-3})); in XLA_TEST_F()
[all …]
Dhlo_test_base.cc329 string_view hlo_string, const absl::optional<ErrorSpec>& error, in RunAndCompare() argument
332 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest()); in RunAndCompare()
342 ::testing::AssertionResult HloTestBase::Run(string_view hlo_string, in Run() argument
347 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest()); in Run()
393 string_view hlo_string, bool run_hlo_passes, in RunMultipleTimes() argument
402 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest()); in RunMultipleTimes()
471 string_view hlo_string, const absl::optional<ErrorSpec>& error, in RunAndCompareNoHloPasses() argument
474 HloRunner::CreateModuleFromString(hlo_string, GetDebugOptionsForTest()); in RunAndCompareNoHloPasses()
Dhlo_test_base.h235 const absl::string_view hlo_string,
239 ::testing::AssertionResult Run(const absl::string_view hlo_string,
244 const absl::string_view hlo_string, bool run_hlo_passes,
252 const absl::string_view hlo_string,
/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()
/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()
Dgpu_alignment_test.cc32 const char* hlo_string = R"( in TEST_F() local
47 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis_test.cc35 const char* hlo_string = R"( in TEST_F() local
58 CompileAndVerifyIr(hlo_string, R"( in TEST_F()

12