/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
D | tree_reduction_rewriter_test.cc | 91 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 130 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 169 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 208 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 248 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 288 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 326 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 366 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
|
D | reduction_vectorization_test.cc | 82 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 130 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 183 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 212 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 262 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 293 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 366 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
|
D | reduction_degenerate_dim_remover_test.cc | 65 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 91 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
|
D | parallel_reduction_test.cc | 71 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F() 129 EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{1e-5, 1e-5})); in TEST_F() 185 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
|
D | reduction_dimension_grouper_test.cc | 61 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
|
D | reduction_layout_normalizer_test.cc | 62 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
|
D | gpu_copy_alone_test.cc | 51 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | elemental_ir_emitter_test.cc | 47 EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{(0.)})); in RunTypeConversionTest() 129 EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{4e-3, 4e-3})); in XLA_TEST_F() 153 EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{(0.)})); in XLA_TEST_F() 176 EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{(0.)})); in XLA_TEST_F() 200 EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{(0.)})); in XLA_TEST_F()
|
D | dynamic_update_slice_test.cc | 79 EXPECT_TRUE(RunAndCompare(std::move(module), fake_argument_ptrs, no_error)); in XLA_TEST_F() 192 RunAndCompare(ParseAndReturnVerifiedModule(kModuleStr).ValueOrDie(), in XLA_TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_fused_conv_rewriter_test.cc | 55 EXPECT_TRUE(RunAndCompare(hlo_with_new_type, ErrorSpec{0.01})) in TestMatchWithAllTypes() 68 EXPECT_TRUE(RunAndCompare(pre_hlo_string, ErrorSpec{0.01})) in TestClamp() 184 EXPECT_TRUE(RunAndCompare(R"( in TEST_F() 366 EXPECT_TRUE(RunAndCompare(kHloString, ErrorSpec{0.01})); in TEST_F()
|
D | horizontal_loop_fusion_test.cc | 229 EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{0, 0})); in TEST_F() 364 EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{1.0e-5, 1.0e-5})); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | dot_operation_test.cc | 1343 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-3, 1e-3})); in XLA_TEST_F() 1358 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-3, 1e-3})); in XLA_TEST_F() 1373 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-3, 1e-3})); in XLA_TEST_F() 1388 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{4e-3, 4e-3})); in XLA_TEST_F() 1409 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{4e-3, 4e-3})); in XLA_TEST_F() 1434 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{4e-3, 4e-3})); in XLA_TEST_F() 1449 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0, 0})); in XLA_TEST_F() 1468 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0, 0})); in XLA_TEST_F() 1484 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0, 0})); in XLA_TEST_F() 1503 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0, 0})); in XLA_TEST_F() [all …]
|
D | reduce_window_test.cc | 1514 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0.001})); in XLA_TEST_F() 1531 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0.001})); in XLA_TEST_F() 1548 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0.001})); in XLA_TEST_F() 1566 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0.001})); in XLA_TEST_F() 1587 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{0.001})); in XLA_TEST_F() 1607 EXPECT_TRUE(RunAndCompare(hlo_string, absl::nullopt)); in XLA_TEST_F() 1627 EXPECT_TRUE(RunAndCompare(hlo_string, absl::nullopt)); in XLA_TEST_F() 1646 EXPECT_TRUE(RunAndCompare(hlo_string, absl::nullopt)); in XLA_TEST_F() 1665 EXPECT_TRUE(RunAndCompare(hlo_string, absl::nullopt)); in XLA_TEST_F() 1684 EXPECT_TRUE(RunAndCompare(hlo_string, absl::nullopt)); in XLA_TEST_F() [all …]
|
D | hlo_test_base.h | 190 ::testing::AssertionResult RunAndCompare( 207 ::testing::AssertionResult RunAndCompare( 227 ::testing::AssertionResult RunAndCompare(
|
D | hlo_test_base.cc | 284 ::testing::AssertionResult HloTestBase::RunAndCompare( in RunAndCompare() function in xla::HloTestBase 312 ::testing::AssertionResult HloTestBase::RunAndCompare( in RunAndCompare() function in xla::HloTestBase 322 return RunAndCompare(std::move(module), fake_argument_ptrs, error, in RunAndCompare() 356 ::testing::AssertionResult HloTestBase::RunAndCompare( in RunAndCompare() function in xla::HloTestBase 365 return RunAndCompare(module_or_status.ConsumeValueOrDie(), error, in RunAndCompare() 542 return RunAndCompare(module_or_status.ConsumeValueOrDie(), error, in RunAndCompareFromFile()
|
D | iota_test.cc | 37 EXPECT_TRUE(RunAndCompare(hlo_text, absl::nullopt)); in XLA_TEST_F() 48 EXPECT_TRUE(RunAndCompare(hlo_text, absl::nullopt)); in XLA_TEST_F()
|
D | get_dimension_size_test.cc | 44 EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{0.01, 0.01})); in TEST_F()
|
D | ptxas_bug_120501638.cc | 78 EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{0.01, 0.01})); in TEST_F()
|
D | sample_text_test.cc | 62 EXPECT_TRUE(RunAndCompare(hlo_string, nullopt)); in TEST_F()
|
D | convolution_test.cc | 1644 EXPECT_TRUE(RunAndCompare(kHlo, ErrorSpec{0.001})); in XLA_TEST_F() 1656 EXPECT_TRUE(RunAndCompare(kHlo, ErrorSpec{0.01, 0.01})); in XLA_TEST_F() 1669 EXPECT_TRUE(RunAndCompare(kHlo, ErrorSpec{0.001})); in XLA_TEST_F() 1683 EXPECT_TRUE(RunAndCompare(kHlo, ErrorSpec{0.001})); in XLA_TEST_F() 1697 EXPECT_TRUE(RunAndCompare(kHlo, ErrorSpec{0.001})); in XLA_TEST_F() 1710 EXPECT_TRUE(RunAndCompare(kHlo, ErrorSpec{0.01, 0.01})); in XLA_TEST_F() 1724 EXPECT_TRUE(RunAndCompare(kHlo, ErrorSpec{0.01, 0.01})); in XLA_TEST_F() 1738 EXPECT_TRUE(RunAndCompare(kHlo, ErrorSpec{0.01, 0.01})); in XLA_TEST_F() 1751 EXPECT_TRUE(RunAndCompare(kHlo, ErrorSpec{0.01, 0.01})); in XLA_TEST_F() 1763 EXPECT_TRUE(RunAndCompare(kHlo, ErrorSpec{0.01, 0.01})); in XLA_TEST_F()
|
D | grouped_convolution_test.cc | 241 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{0.01, 0.01}, in XLA_TEST_P() 275 EXPECT_TRUE(RunAndCompare(std::move(module), fake_argument_ptrs, in XLA_TEST_F()
|
D | conv_depthwise_test.cc | 99 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{0.01, 0.01}, in XLA_TEST_P()
|
D | reduce_test.cc | 1036 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-5, 1e-5})); in XLA_TEST_F() 1066 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-5, 1e-5})); in XLA_TEST_F() 1094 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-5, 1e-5})); in XLA_TEST_F() 1122 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-5, 1e-5})); in XLA_TEST_F() 1159 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-5, 1e-5})); in XLA_TEST_F() 1191 EXPECT_TRUE(RunAndCompare(hlo_string, ErrorSpec{1e-5, 1e-5})); in XLA_TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/tools/ |
D | run_hlo_module.h | 65 Status RunAndCompare(
|
D | run_hlo_module_main.cc | 160 xla::RunAndCompare(hlo_filename, test_platform_name, in main()
|