| /external/tensorflow/tensorflow/compiler/xla/service/ |
| while_loop_analysis_test.cc | 53 ParseAndReturnVerifiedModule(kHloModule));
84 ParseAndReturnVerifiedModule(kHloModule));
117 ParseAndReturnVerifiedModule(kHloModule));
|
| sort_simplifier_test.cc | 53 ParseAndReturnVerifiedModule(hlo_string));
91 ParseAndReturnVerifiedModule(hlo_string));
122 ParseAndReturnVerifiedModule(hlo_string));
148 ParseAndReturnVerifiedModule(hlo_string));
|
| while_loop_trip_count_annotator_test.cc | 53 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr));
89 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr));
125 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr));
161 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr));
200 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr));
|
| batch_dot_simplification_test.cc | 41 ParseAndReturnVerifiedModule(hlo_text));
65 ParseAndReturnVerifiedModule(hlo_text));
89 ParseAndReturnVerifiedModule(hlo_text));
113 ParseAndReturnVerifiedModule(hlo_text));
137 ParseAndReturnVerifiedModule(hlo_text));
161 ParseAndReturnVerifiedModule(hlo_text));
188 ParseAndReturnVerifiedModule(hlo_text));
209 ParseAndReturnVerifiedModule(hlo_text));
|
| hlo_element_type_converter_test.cc | 42 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
58 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
77 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
105 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
129 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
155 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
187 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
|
| while_loop_simplifier_test.cc | 88 return ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
126 return ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
250 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
282 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
311 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
334 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
362 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
404 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
452 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie();
485 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie()
[all...] |
| ar_crs_combiner_test.cc | 42 ParseAndReturnVerifiedModule(module_str));
62 ParseAndReturnVerifiedModule(module_str));
81 ParseAndReturnVerifiedModule(module_str));
102 ParseAndReturnVerifiedModule(module_str));
122 ParseAndReturnVerifiedModule(module_str));
142 ParseAndReturnVerifiedModule(module_str));
164 ParseAndReturnVerifiedModule(module_str));
186 ParseAndReturnVerifiedModule(module_str));
209 ParseAndReturnVerifiedModule(module_str));
245 ParseAndReturnVerifiedModule(module_str))
[all...] |
| elemental_ir_emitter_test.cc | 35 ParseAndReturnVerifiedModule(hlo_text, config));
93 ParseAndReturnVerifiedModule(hlo_text, config));
|
| stable_sort_expander_test.cc | 103 ParseAndReturnVerifiedModule(hlo_string));
149 ParseAndReturnVerifiedModule(hlo_string));
179 ParseAndReturnVerifiedModule(hlo_string));
215 ParseAndReturnVerifiedModule(hlo_string));
242 ParseAndReturnVerifiedModule(hlo_string));
279 ParseAndReturnVerifiedModule(hlo_string));
315 ParseAndReturnVerifiedModule(hlo_string));
346 ParseAndReturnVerifiedModule(hlo_string));
|
| dfs_hlo_visitor_with_default_test.cc | 83 ParseAndReturnVerifiedModule(hlo_string).ConsumeValueOrDie();
|
| hlo_pass_pipeline_test.cc | 36 ParseAndReturnVerifiedModule(hlo_string));
110 ParseAndReturnVerifiedModule(module_str));
133 ParseAndReturnVerifiedModule(module_str));
195 ParseAndReturnVerifiedModule(module_str));
247 ParseAndReturnVerifiedModule(module_str));
|
| hlo_domain_test.cc | 140 ParseAndReturnVerifiedModule(hlo_string));
183 ParseAndReturnVerifiedModule(hlo_string));
211 ParseAndReturnVerifiedModule(hlo_string));
249 ParseAndReturnVerifiedModule(hlo_string));
273 ParseAndReturnVerifiedModule(hlo_string));
305 ParseAndReturnVerifiedModule(hlo_string));
361 ParseAndReturnVerifiedModule(hlo_string));
449 ParseAndReturnVerifiedModule(hlo_string));
493 ASSERT_TRUE(ParseAndReturnVerifiedModule(hlo_string).status().ok());
511 ParseAndReturnVerifiedModule(hlo_string))
[all...] |
| hlo_constant_folding_test.cc | 227 ParseAndReturnVerifiedModule(kConstantFoldReduce));
240 ParseAndReturnVerifiedModule(kConstantFoldReduce));
262 ParseAndReturnVerifiedModule(kConstantFoldLargePad));
284 ParseAndReturnVerifiedModule(kModuleStr));
310 ParseAndReturnVerifiedModule(kModuleStr));
|
| hlo_evaluator_test.cc | [all...] |
| layout_assignment_test.cc | 680 ParseAndReturnVerifiedModule(module_str));
[all...] |
| batchnorm_expander_test.cc | 145 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(module_str));
|
| hlo_graph_dumper_test.cc | 157 ParseAndReturnVerifiedModule(hlo_string));
|
| hlo_cse_test.cc | 282 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_string));
332 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_string));
374 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_string));
415 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_string));
685 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_string));
736 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_string));
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| parallel_task_assignment_test.cc | 61 ParseAndReturnVerifiedModule(hlo_string));
89 ParseAndReturnVerifiedModule(hlo_string));
106 ParseAndReturnVerifiedModule(hlo_string));
123 ParseAndReturnVerifiedModule(hlo_string));
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| ptxas_bug_120501638.cc | 77 ParseAndReturnVerifiedModule(kModuleStr, config));
|
| multioutput_fusion_test.cc | 211 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie();
242 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie();
272 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie();
312 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie();
341 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie();
371 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie();
401 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie();
432 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie();
465 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie();
498 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie()
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_literal_caching_test.cc | 64 ParseAndReturnVerifiedModule(hlo_text));
112 ParseAndReturnVerifiedModule(hlo_text));
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_conv_pad_for_tensor_cores_test.cc | 35 auto module = ParseAndReturnVerifiedModule(R"(
60 auto module = ParseAndReturnVerifiedModule(R"(
83 auto module = ParseAndReturnVerifiedModule(R"(
103 auto module = ParseAndReturnVerifiedModule(R"(
125 auto module = ParseAndReturnVerifiedModule(R"(
147 auto module = ParseAndReturnVerifiedModule(R"(
169 auto module = ParseAndReturnVerifiedModule(R"(
|
| variadic_op_splitter_test.cc | 38 auto module = ParseAndReturnVerifiedModule(R"(
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| hlo_extractor_test.cc | 40 ParseAndReturnVerifiedModule(hlo_string));
78 ParseAndReturnVerifiedModule(hlo_string));
122 ParseAndReturnVerifiedModule(hlo_string));
|
