/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_fusible_test.cc | 38 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 58 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 96 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 133 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 157 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 194 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 221 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 238 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 256 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 275 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() [all …]
|
D | multi_output_fusion_test.cc | 63 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 90 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 117 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 146 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 181 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 214 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 244 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 276 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 294 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() 319 auto module = ParseAndReturnVerifiedModule(absl::StrCat(kModulePrefix, R"( in TEST_F() [all …]
|
D | fusion_merger_test.cc | 47 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 113 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 161 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 204 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 238 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 286 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 364 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 419 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 450 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 473 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() [all …]
|
D | cublas_gemm_pad_for_tensor_cores_test.cc | 35 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 70 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 138 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 171 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 185 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 200 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 215 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 230 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 255 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F()
|
D | cudnn_pad_for_convolutions_test.cc | 35 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 60 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 83 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 103 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 125 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 147 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 169 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 194 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 220 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 241 auto module = ParseAndReturnVerifiedModule(R"(
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
D | gpu_kernel_tiling_test.cc | 64 ParseAndReturnVerifiedModule(kHloString, ConfigWithLayoutAssignment()) in TEST_F() 98 ParseAndReturnVerifiedModule(kHloString, ConfigWithLayoutAssignment()) in TEST_F() 147 ParseAndReturnVerifiedModule(kHloString, ConfigWithoutLayoutAssignment()) in TEST_F() 187 ParseAndReturnVerifiedModule(kHloString, ConfigWithoutLayoutAssignment()) in TEST_F() 228 ParseAndReturnVerifiedModule(kHloString, ConfigWithoutLayoutAssignment()) in TEST_F() 261 ParseAndReturnVerifiedModule(kHloString, ConfigWithoutLayoutAssignment()) in TEST_F() 294 ParseAndReturnVerifiedModule(kHloString, ConfigWithoutLayoutAssignment()) in TEST_F() 335 ParseAndReturnVerifiedModule(kHloString, ConfigWithoutLayoutAssignment()) in TEST_F() 372 ParseAndReturnVerifiedModule(kHloString, ConfigWithoutLayoutAssignment()) in TEST_F() 436 ParseAndReturnVerifiedModule(kHloString, ConfigWithoutLayoutAssignment()) in TEST_F() [all …]
|
D | gpu_unrolling_test.cc | 52 ParseAndReturnVerifiedModule(kAddModule, config).ValueOrDie(); in TEST_F() 72 ParseAndReturnVerifiedModule(kAddModule, config).ValueOrDie(); in TEST_F() 92 ParseAndReturnVerifiedModule(kAddModule, config).ValueOrDie(); in TEST_F() 124 ParseAndReturnVerifiedModule(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() 155 ParseAndReturnVerifiedModule(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() 189 ParseAndReturnVerifiedModule(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() 223 ParseAndReturnVerifiedModule(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() 248 ParseAndReturnVerifiedModule(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() 286 ParseAndReturnVerifiedModule(kMultiOutputFusionModule, config) in TEST_F()
|
D | reduction_vectorization_test.cc | 54 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 102 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 150 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 203 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 234 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 284 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 338 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
|
D | gpu_index_test.cc | 70 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 95 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 126 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 161 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 190 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | ar_crs_combiner_test.cc | 42 ParseAndReturnVerifiedModule(module_str)); in TEST_F() 62 ParseAndReturnVerifiedModule(module_str)); in TEST_F() 81 ParseAndReturnVerifiedModule(module_str)); in TEST_F() 102 ParseAndReturnVerifiedModule(module_str)); in TEST_F() 122 ParseAndReturnVerifiedModule(module_str)); in TEST_F() 142 ParseAndReturnVerifiedModule(module_str)); in TEST_F() 164 ParseAndReturnVerifiedModule(module_str)); in TEST_F() 186 ParseAndReturnVerifiedModule(module_str)); in TEST_F() 209 ParseAndReturnVerifiedModule(module_str)); in TEST_F() 245 ParseAndReturnVerifiedModule(module_str)); in TEST_F() [all …]
|
D | sharding_propagation_test.cc | 150 ParseAndReturnVerifiedModule(hlo_string)); in TEST_P() 182 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 205 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 228 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 251 ParseAndReturnVerifiedModule(hlo_string)); in TEST_P() 282 ParseAndReturnVerifiedModule(hlo_string)); in TEST_P() 313 ParseAndReturnVerifiedModule(hlo_string)); in TEST_P() 343 ParseAndReturnVerifiedModule(hlo_string)); in TEST_P() 373 ParseAndReturnVerifiedModule(hlo_string)); in TEST_P() 406 ParseAndReturnVerifiedModule(hlo_string)); in TEST_P() [all …]
|
D | while_loop_analysis_test.cc | 53 ParseAndReturnVerifiedModule(kHloModule)); in TEST_F() 84 ParseAndReturnVerifiedModule(kHloModule)); in TEST_F() 117 ParseAndReturnVerifiedModule(kHloModule)); in TEST_F() 152 ParseAndReturnVerifiedModule(kHloModule)); in TEST_F() 191 ParseAndReturnVerifiedModule(kHloModule)); in TEST_F() 230 ParseAndReturnVerifiedModule(kHloModule)); in TEST_F() 265 ParseAndReturnVerifiedModule(kHloModule)); in TEST_F()
|
D | while_loop_constant_sinking_test.cc | 57 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 97 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 138 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 186 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 228 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 274 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 314 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 357 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 408 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
|
D | batch_dot_simplification_test.cc | 41 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 65 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 89 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 113 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 137 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 161 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 188 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F() 209 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
|
D | hlo_module_dce_test.cc | 85 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 122 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 162 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 206 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 248 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 316 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 382 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 419 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F() 454 auto module = ParseAndReturnVerifiedModule(R"( in TEST_F()
|
D | conditional_code_motion_test.cc | 73 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 122 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 158 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 199 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 240 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 276 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 345 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 401 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 470 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 527 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() [all …]
|
D | while_loop_simplifier_test.cc | 88 return ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in MakeModuleWithSimpleLoop() 126 return ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in MakeModuleWithSimpleLoopTupleElementLoopBound() 263 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 295 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 324 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 347 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 375 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 417 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 472 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 506 auto m = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() [all …]
|
D | slice_sinker_test.cc | 64 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 104 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 150 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 199 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 239 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 261 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 283 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 305 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 327 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 351 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() [all …]
|
D | memory_space_propagation_test.cc | 65 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 68 TF_ASSERT_OK_AND_ASSIGN(auto ref, ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 129 ParseAndReturnVerifiedModule(expected_hlo_string)); in TEST_F() 198 ParseAndReturnVerifiedModule(expected_hlo_string)); in TEST_F() 272 ParseAndReturnVerifiedModule(expected_hlo_string)); in TEST_F() 346 ParseAndReturnVerifiedModule(expected_hlo_string)); in TEST_F()
|
D | hlo_element_type_converter_test.cc | 42 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 58 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 77 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 105 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 129 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 155 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 187 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F()
|
D | while_loop_trip_count_annotator_test.cc | 53 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 89 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 125 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 161 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F() 200 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
|
D | while_loop_expensive_invariant_code_motion_test.cc | 66 auto m = ParseAndReturnVerifiedModule(kModuleWithNonInflatingInvariantDot) in TEST_F() 87 auto m = ParseAndReturnVerifiedModule(kModuleWithNonInflatingInvariantDot) in TEST_F() 108 auto m = ParseAndReturnVerifiedModule(kModuleWithNonInflatingInvariantDot) in TEST_F() 156 auto m = ParseAndReturnVerifiedModule(kModuleWithInflatingInvariantDot) in TEST_F() 171 auto m = ParseAndReturnVerifiedModule(kModuleWithInflatingInvariantDot) in TEST_F() 223 ParseAndReturnVerifiedModule(kModuleWithDuplicateOperands).ValueOrDie(); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | parallel_task_assignment_test.cc | 61 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 89 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 106 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 123 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 168 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 189 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F() 203 ParseAndReturnVerifiedModule(hlo_string)); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | multioutput_fusion_test.cc | 213 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie(); in XLA_TEST_F() 244 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie(); in XLA_TEST_F() 274 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie(); in XLA_TEST_F() 314 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie(); in XLA_TEST_F() 336 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie(); in XLA_TEST_F() 359 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie(); in XLA_TEST_F() 382 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie(); in XLA_TEST_F() 405 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie(); in XLA_TEST_F() 436 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie(); in XLA_TEST_F() 459 auto module = ParseAndReturnVerifiedModule(testcase).ValueOrDie(); in XLA_TEST_F() [all …]
|
D | test_utils_test.cc | 60 auto module = ParseAndReturnVerifiedModule( in XLA_TEST_F() 88 auto module = ParseAndReturnVerifiedModule( in XLA_TEST_F() 120 auto module = ParseAndReturnVerifiedModule(R"( in XLA_TEST_F() 151 auto module = ParseAndReturnVerifiedModule(R"( in XLA_TEST_F() 182 auto module = ParseAndReturnVerifiedModule(R"( in XLA_TEST_F() 212 auto module = ParseAndReturnVerifiedModule(R"( in XLA_TEST_F() 240 auto module = ParseAndReturnVerifiedModule(R"( in XLA_TEST_F() 274 auto module = ParseAndReturnVerifiedModule(R"( in XLA_TEST_F()
|