Home
last modified time | relevance | path

Searched refs:ParseHloString (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible_test.cc38 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
58 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
95 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
119 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
156 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
182 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
199 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
216 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
235 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
254 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
[all …]
Dmulti_output_fusion_test.cc50 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
77 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
104 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
133 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
168 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
201 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
231 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
262 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
280 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
304 auto module = ParseHloString(absl::StrCat(kModulePrefix, R"( in TEST_F()
[all …]
Dfusion_merger_test.cc43 auto module = ParseHloString(R"( in TEST_F()
109 auto module = ParseHloString(R"( in TEST_F()
157 auto module = ParseHloString(R"( in TEST_F()
200 auto module = ParseHloString(R"( in TEST_F()
234 auto module = ParseHloString(R"( in TEST_F()
268 auto module = ParseHloString(R"( in TEST_F()
Dinstruction_fusion_test.cc147 auto module = ParseHloString(R"( in TEST_F()
176 auto module = ParseHloString(R"( in TEST_F()
199 auto module = ParseHloString(R"( in TEST_F()
229 auto module = ParseHloString(R"( in TEST_F()
248 auto module = ParseHloString(R"( in TEST_F()
270 auto module = ParseHloString(R"( in TEST_F()
292 auto module = ParseHloString(R"( in TEST_F()
308 auto module = ParseHloString(R"( in TEST_F()
335 auto module = ParseHloString(R"( in TEST_F()
387 auto module = ParseHloString(R"( in TEST_F()
[all …]
Dgpu_sanitize_constant_names_test.cc41 ParseHloString(kHloString)); in TEST_F()
56 ParseHloString(kHloString)); in TEST_F()
73 ParseHloString(kHloString)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_kernel_tiling_test.cc64 ParseHloString(kHloString, ConfigWithLayoutAssignment()).ValueOrDie(); in TEST_F()
90 ParseHloString(kHloString, ConfigWithLayoutAssignment()).ValueOrDie(); in TEST_F()
117 ParseHloString(kHloString, ConfigWithoutLayoutAssignment()).ValueOrDie(); in TEST_F()
151 ParseHloString(kHloString, ConfigWithoutLayoutAssignment()).ValueOrDie(); in TEST_F()
186 ParseHloString(kHloString, ConfigWithoutLayoutAssignment()).ValueOrDie(); in TEST_F()
213 ParseHloString(kHloString, ConfigWithoutLayoutAssignment()).ValueOrDie(); in TEST_F()
240 ParseHloString(kHloString, ConfigWithoutLayoutAssignment()).ValueOrDie(); in TEST_F()
275 ParseHloString(kHloString, ConfigWithoutLayoutAssignment()).ValueOrDie(); in TEST_F()
306 ParseHloString(kHloString, ConfigWithoutLayoutAssignment()).ValueOrDie(); in TEST_F()
352 ParseHloString(kHloString, ConfigWithoutLayoutAssignment()).ValueOrDie(); in TEST_F()
[all …]
Dgpu_unrolling_test.cc51 auto hlo_module = ParseHloString(kAddModule, config).ValueOrDie(); in TEST_F()
70 auto hlo_module = ParseHloString(kAddModule, config).ValueOrDie(); in TEST_F()
89 auto hlo_module = ParseHloString(kAddModule, config).ValueOrDie(); in TEST_F()
120 auto hlo_module = ParseHloString(kUnfusedAddModule, config).ValueOrDie(); in TEST_F()
164 ParseHloString(kMultiOutputFusionModule, config).ValueOrDie(); in TEST_F()
Dgpu_index_test.cc70 auto module = ParseHloString(R"( in TEST_F()
94 auto module = ParseHloString(R"( in TEST_F()
124 auto module = ParseHloString(R"( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier_test.cc169 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
198 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
217 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
235 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
254 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
273 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
290 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(hlo_string)); in TEST_F()
355 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kConvHloString)); in TEST_F()
366 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kConvHloString)); in TEST_F()
386 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kAddWithLayoutChangeHlo)); in TEST_F()
[all …]
Dhlo_module_dce_test.cc74 auto module = ParseHloString(R"( in TEST_F()
111 auto module = ParseHloString(R"( in TEST_F()
151 auto module = ParseHloString(R"( in TEST_F()
194 auto module = ParseHloString(R"( in TEST_F()
236 auto module = ParseHloString(R"( in TEST_F()
304 auto module = ParseHloString(R"( in TEST_F()
370 auto module = ParseHloString(R"( in TEST_F()
407 auto module = ParseHloString(R"( in TEST_F()
Dwhile_loop_constant_sinking_test.cc58 ParseHloString(hlo_string)); in TEST_F()
98 ParseHloString(hlo_string)); in TEST_F()
139 ParseHloString(hlo_string)); in TEST_F()
187 ParseHloString(hlo_string)); in TEST_F()
229 ParseHloString(hlo_string)); in TEST_F()
275 ParseHloString(hlo_string)); in TEST_F()
315 ParseHloString(hlo_string)); in TEST_F()
358 ParseHloString(hlo_string)); in TEST_F()
409 ParseHloString(hlo_string)); in TEST_F()
Dhlo_liveness_analysis_test.cc62 auto module = ParseHloString(R"( in TEST_F()
78 auto module = ParseHloString(R"( in TEST_F()
97 auto module = ParseHloString(R"( in TEST_F()
116 auto module = ParseHloString(R"( in TEST_F()
143 auto module = ParseHloString(R"( in TEST_F()
165 auto module = ParseHloString(R"( in TEST_F()
202 auto module = ParseHloString(R"( in TEST_F()
243 auto module = ParseHloString(R"( in TEST_F()
294 auto module = ParseHloString(R"( in TEST_F()
348 auto module = ParseHloString(R"( in TEST_F()
[all …]
Dinstruction_fusion_test.cc50 auto module = ParseHloString(R"( in TEST_F()
70 auto module = ParseHloString(R"( in TEST_F()
93 auto module = ParseHloString(R"( in TEST_F()
199 auto module = ParseHloString(R"( in TEST_F()
224 auto module = ParseHloString(R"( in TEST_F()
255 module = ParseHloString(R"( in TEST_F()
287 module = ParseHloString(R"( in TEST_F()
320 module = ParseHloString(R"( in TEST_F()
398 auto module = ParseHloString(R"( in TEST_F()
423 auto module = ParseHloString(R"( in TEST_F()
[all …]
Dhlo_module_group_test.cc48 ParseHloString(text)); in TEST_F()
87 ParseHloString(text_0)); in TEST_F()
89 ParseHloString(text_1)); in TEST_F()
126 ParseHloString(text_0)); in TEST_F()
128 ParseHloString(text_1)); in TEST_F()
181 ParseHloString(absl::StrFormat(text, i, send_channel, send_channel, in TEST_F()
Dhlo_input_output_alias_config_test.cc83 ParseHloString(module_str)); in TEST_F()
115 ParseHloString(module_str)); in TEST_F()
154 ParseHloString(module_str)); in TEST_F()
185 ParseHloString(module_str)); in TEST_F()
211 ParseHloString(module_str)); in TEST_F()
Dhlo_get_dimension_size_rewriter_test.cc43 auto module = ParseHloString(R"( in TEST_F()
59 auto module = ParseHloString(R"( in TEST_F()
71 auto module = ParseHloString(R"( in TEST_F()
Dhlo_parser.h35 StatusOr<std::unique_ptr<HloModule>> ParseHloString(
41 Status ParseHloString(absl::string_view str, HloModule* module);
45 StatusOr<std::unique_ptr<HloModule>> ParseHloString(absl::string_view str);
Dhlo_schedule_test.cc56 ParseHloString(module_str)); in TEST_F()
90 ParseHloString(module_str)); in TEST_F()
139 ParseHloString(module_str)); in TEST_F()
183 ParseHloString(module_str)); in TEST_F()
244 ParseHloString(module_str)); in TEST_F()
313 ParseHloString(module_str)); in TEST_F()
Dgather_expander_test.cc39 ParseHloString(hlo_text)); in TEST()
66 ParseHloString(hlo_text)); in TEST()
124 ParseHloString(hlo_text)); in TEST()
Ddynamic_parameter_binding_test.cc59 ParseHloString(module_str)); in TEST_F()
97 ParseHloString(module_str)); in TEST_F()
136 ParseHloString(module_str)); in TEST_F()
Dconvolution_group_converter_test.cc47 ParseHloString(hlo_string)); in TEST_F()
78 ParseHloString(hlo_string)); in TEST_F()
107 ParseHloString(hlo_string)); in TEST_F()
Ddynamic_index_splitter_test.cc48 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kDynamicSlice, config)); in TEST_F()
86 ParseHloString(kDynamicUpdateSlice, config)); in TEST_F()
124 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseHloString(kDynamicSlice, config)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils_test.cc60 auto module = ParseHloString( in XLA_TEST_F()
78 auto module = ParseHloString( in XLA_TEST_F()
106 auto module = ParseHloString( in XLA_TEST_F()
138 auto module = ParseHloString(R"( in XLA_TEST_F()
169 auto module = ParseHloString(R"( in XLA_TEST_F()
200 auto module = ParseHloString(R"( in XLA_TEST_F()
230 auto module = ParseHloString(R"( in XLA_TEST_F()
258 auto module = ParseHloString(R"( in XLA_TEST_F()
292 auto module = ParseHloString(R"( in XLA_TEST_F()
Dall_reduce_test.cc47 ParseHloString(module_str, GetModuleConfigForTest()).ValueOrDie(); in XLA_TEST_F()
68 ParseHloString(module_str, GetModuleConfigForTest()).ValueOrDie(); in XLA_TEST_F()
94 ParseHloString(module_str, GetModuleConfigForTest()).ValueOrDie(); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_outfeed_test.cc45 ParseHloString(hlo_text)); in TEST_F()
73 ParseHloString(hlo_text)); in TEST_F()

123