/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | dynamic_dimension_simplifier_test.cc | 53 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 D | slice_sinker_test.cc | 45 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 D | scatter_simplifier_test.cc | 33 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 D | gather_simplifier_test.cc | 32 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 D | hlo_verifier_test.cc | 1521 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 D | scatter_expander_test.cc | 46 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 D | pattern_matcher_test.cc | 34 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 D | while_loop_trip_count_annotator_test.cc | 31 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 D | algebraic_simplifier_test.cc | 80 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 D | real_imag_expander_test.cc | 42 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 D | all_reduce_simplifier_test.cc | 42 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 D | simplify_fp_conversions_test.cc | 35 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 D | change_op_data_type_test.cc | 41 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 D | select_and_scatter_expander_test.cc | 27 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 D | hlo_constant_folding_test.cc | 275 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 D | dynamic_update_slice_test.cc | 34 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 D | logistic_expander_test.cc | 54 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 D | tuple_simplifier_test.cc | 290 const char* kModuleStr = R"( in TEST_F() local 299 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
|
H A D | hlo_cse_test.cc | 918 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 D | collective_ops_test.cc | 485 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 D | custom_call_test.cc | 216 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 D | get_dimension_size_test.cc | 28 const char* const kModuleStr = R"( in TEST_F() local 43 ParseAndReturnVerifiedModule(kModuleStr)); in TEST_F()
|
H A D | ptxas_bug_120501638.cc | 30 const char* const kModuleStr = R"( in TEST_F() local 77 ParseAndReturnVerifiedModule(kModuleStr, config)); in TEST_F()
|