/external/tensorflow/tensorflow/compiler/xla/ |
D | reference_util_test.cc | 58 actual_literal, ErrorSpec(0.0001)); in TEST_F() 70 actual_literal, ErrorSpec(0.0001)); in TEST_F() 78 ErrorSpec(0.0001)); in TEST_F() 86 ErrorSpec(0.0001)); in TEST_F() 101 ErrorSpec(0.0001)); in TEST_F() 111 actual_literal, ErrorSpec(0.0001)); in TEST_F() 125 ErrorSpec(0.0001)); in TEST_F() 142 ErrorSpec(0.0001)); in TEST_F() 150 ErrorSpec(0.0001)); in TEST_F() 158 ErrorSpec(0.0001)); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | literal_test_util.h | 95 const ErrorSpec& error_spec, 102 const ErrorSpec& error); 106 const LiteralSlice& actual, const ErrorSpec& error); 111 const LiteralSlice& actual, const ErrorSpec& error); 118 const LiteralSlice& actual, const ErrorSpec& error); 125 const LiteralSlice& actual, const ErrorSpec& error); 132 const ErrorSpec& error); 137 const ErrorSpec& error); 142 const ErrorSpec& error); 149 const absl::optional<ErrorSpec>& error) TF_MUST_USE_RESULT; [all …]
|
D | broadcast_simple_test.cc | 111 ComputeAndCompareR0<float>(&b, 1.5, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 118 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 131 ErrorSpec(0.0001)); in XLA_TEST_F() 138 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 145 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 159 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001)); 172 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001)); 185 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001)); 203 ComputeAndCompareR3<float>(&b, expected, {}, ErrorSpec(0.0001)); 221 ComputeAndCompareR3<float>(&b, expected, {}, ErrorSpec(0.0001)); [all …]
|
D | concat_test.cc | 56 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 65 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 88 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 98 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 108 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 118 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 137 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 148 ErrorSpec(0.0001)); in XLA_TEST_F() 164 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 178 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() [all …]
|
D | triangular_solve_test.cc | 136 ErrorSpec(1e-2, 1e-2)); in XLA_TEST_F() 157 ErrorSpec(1e-2, 1e-2)); in XLA_TEST_F() 178 ErrorSpec(1e-2, 1e-2)); in XLA_TEST_F() 199 ErrorSpec(1e-2, 1e-2)); in XLA_TEST_F() 221 ErrorSpec(1e-2, 1e-2)); in XLA_TEST_F() 243 ErrorSpec(1e-2, 1e-2)); in XLA_TEST_F() 262 ErrorSpec(1e-2, 1e-2)); in XLA_TEST_F() 284 ErrorSpec(1e-2, 1e-2)); in XLA_TEST_F() 306 ErrorSpec(1e-2, 1e-2)); in XLA_TEST_F() 328 ErrorSpec(1e-2, 1e-2)); in XLA_TEST_F() [all …]
|
D | hlo_test_base.h | 205 const absl::optional<ErrorSpec>& error, 214 const absl::optional<ErrorSpec>& error, 220 std::unique_ptr<HloModule> module, const absl::optional<ErrorSpec>& error, 227 std::unique_ptr<HloModule> module, const absl::optional<ErrorSpec>& error, 236 const absl::optional<ErrorSpec>& error, 248 const string& filename, const absl::optional<ErrorSpec>& error, 253 const absl::optional<ErrorSpec>& error, 257 const string& filename, const absl::optional<ErrorSpec>& error, 318 ErrorSpec error_spec_{0.0001}; 334 const absl::optional<ErrorSpec>& error, bool run_hlo_passes,
|
D | select_and_scatter_test.cc | 87 ComputeAndCompare(&builder_, {}, ErrorSpec(1e-5)); in XLA_TEST_P() 211 ComputeAndCompareR1<float>(&builder_, {}, {}, ErrorSpec(1e-7)); in XLA_TEST_F() 223 ComputeAndCompareR1<float>(&builder_, expected, {}, ErrorSpec(1e-7)); in XLA_TEST_F() 272 ComputeAndCompareR2<float>(&builder_, expected, {}, ErrorSpec(1e-7)); in XLA_TEST_F() 337 ComputeAndCompareR2<float>(&builder_, expected, {}, ErrorSpec(1e-7)); in XLA_TEST_F() 362 ComputeAndCompareR4<float>(&builder_, e, {}, ErrorSpec(1e-7)); in TEST_F() 387 ComputeAndCompareR4<float>(&builder_, e, {}, ErrorSpec(1e-7)); in TEST_F() 412 ComputeAndCompareR4<float>(&builder_, e, {}, ErrorSpec(1e-7)); in TEST_F() 435 ComputeAndCompareR4<float>(&builder_, *e, {}, ErrorSpec(1e-7)); in TEST_F() 445 ComputeAndCompareR1<float>(&builder_, expected, {}, ErrorSpec(1e-7)); in XLA_TEST_F() [all …]
|
D | binop_scaling_test.cc | 44 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001)); in TEST_F() 60 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001)); in TEST_F() 76 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001)); in TEST_F() 92 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001)); in TEST_F() 109 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001)); in TEST_F()
|
D | params_test.cc | 52 ErrorSpec(0.0001f)); in XLA_TEST_F() 64 ErrorSpec(0.01f)); in XLA_TEST_F() 76 ErrorSpec(0.01f)); in XLA_TEST_F() 103 {param0_data.get()}, ErrorSpec(0.01f)); in XLA_TEST_F() 118 ErrorSpec(0.01f)); in XLA_TEST_F() 149 ErrorSpec(0.0001f)); in XLA_TEST_F() 181 ErrorSpec(0.0001f)); in XLA_TEST_F() 209 ErrorSpec(0.0001f)); in XLA_TEST_F() 246 ComputeAndCompareR1<float>(&builder, sum, param_data, ErrorSpec(0.0001f)); in XLA_TEST_F() 281 ComputeAndCompareR0<float>(&builder, target, param_data, ErrorSpec(0.0001f)); in XLA_TEST_F() [all …]
|
D | reduce_test.cc | 116 ErrorSpec(0.001)); in RunR1ToR0Test() 225 ErrorSpec(0.01, 1e-4)); in RunR2ToR0Test() 254 ErrorSpec(0.01, 1e-4)); in RunR2ToR1Test() 264 ErrorSpec(0.01, 1e-4)); in ComputeAndCompareGeneric() 464 ErrorSpec(0.01, 1e-4)); in XLA_TEST_F() 495 ErrorSpec(0.01, 1e-4)); in XLA_TEST_F() 511 ComputeAndCompare(&builder, {std::move(input_data)}, ErrorSpec(0.01, 1e-4)); in XLA_TEST_F() 543 ErrorSpec(0.01, 1e-4)); in XLA_TEST_F() 569 ComputeAndCompareR0<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 581 ComputeAndCompareR0<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() [all …]
|
D | cholesky_test.cc | 62 ErrorSpec(1e-4, 1e-4)); in XLA_TEST_F() 88 ErrorSpec(1e-4, 1e-4)); in XLA_TEST_F() 111 ErrorSpec(1e-4, 1e-4)); in XLA_TEST_F() 150 ErrorSpec(1e-4, 1e-4)); in XLA_TEST_F() 193 ErrorSpec(1e-4, 1e-4)); in XLA_TEST_P()
|
D | literal_test_util_test.cc | 63 ErrorSpec error(0.001); in TEST() 118 EXPECT_TRUE(LiteralTestUtil::Near(a, b, ErrorSpec{0.0001})); in TEST() 126 EXPECT_TRUE(LiteralTestUtil::Near(a, b, ErrorSpec{0.0001})); in TEST() 134 EXPECT_FALSE(LiteralTestUtil::Near(a, b, ErrorSpec{0.0001})); in TEST() 135 EXPECT_FALSE(LiteralTestUtil::Near(b, a, ErrorSpec{0.0001})); in TEST()
|
D | call_test.cc | 83 ComputeAndCompareR0<float>(&builder, 42.0, {}, ErrorSpec(0.01f)); in XLA_TEST_F() 93 ComputeAndCompareR1<float>(&builder, {}, {}, ErrorSpec(0.01f)); in XLA_TEST_F() 105 ComputeAndCompareR1<float>(&builder, {3.0f, 5.0f}, {}, ErrorSpec(0.01f)); in XLA_TEST_F() 136 ComputeAndCompareR0<float>(&builder3, 10.0f, {start.get()}, ErrorSpec(0.0f)); in XLA_TEST_F() 146 ComputeAndCompareTuple(&builder, tuple, {}, ErrorSpec(0.01f)); in XLA_TEST_F()
|
D | map_test.cc | 180 ErrorSpec(0.01f)); in TEST_F() 194 ErrorSpec(0.01f)); in XLA_TEST_F() 209 {param0_data.get()}, ErrorSpec(0.01f)); in TEST_F() 251 {param0_data.get()}, ErrorSpec(0.01f)); in TEST_F() 267 ErrorSpec(0.01f)); in XLA_TEST_F() 284 {param0_data.get()}, ErrorSpec(0.01f)); in TEST_F() 301 ErrorSpec(0.01f)); in TEST_F() 339 ComputeAndCompareR0<float>(&builder, 73.0, {}, ErrorSpec(0.01f)); in XLA_TEST_F() 361 ErrorSpec(0.01f)); in TEST_F() 437 ErrorSpec(0.01f)); in TEST_F() [all …]
|
D | client_library_test_base.h | 132 ErrorSpec error); 142 ErrorSpec error); 158 ErrorSpec error); 168 ErrorSpec error); 178 ErrorSpec error); 188 ErrorSpec error, 205 absl::Span<GlobalData* const> arguments, ErrorSpec error, 219 ErrorSpec error); 226 absl::Span<const Literal> arguments, ErrorSpec error); 429 absl::Span<GlobalData* const> arguments, ErrorSpec error) { in ComputeAndCompareR0() [all …]
|
D | matrix_ops_simple_test.cc | 70 this->ComputeAndCompareLiteral(&builder, expected, {}, ErrorSpec(1e-5)); in XLA_TYPED_TEST() 98 this->ComputeAndCompareLiteral(&builder, expected, {}, ErrorSpec(1e-5)); in XLA_TYPED_TEST() 117 this->ComputeAndCompareLiteral(&builder, expected, {}, ErrorSpec(1e-6)); in XLA_TYPED_TEST() 150 ErrorSpec error_spec(1e-6); in TestImpl() 152 error_spec = ErrorSpec(1e-6, 2e-4); in TestImpl() 237 ErrorSpec(1e-6)); in TestImpl()
|
D | bfloat16_test.cc | 49 const ErrorSpec error_spec_{0.001, 0.001}; 68 ErrorSpec(0.01, 0.01)); in XLA_TEST_F() 112 ComputeAndCompareTuple(&builder, expected, {}, ErrorSpec(0.01, 0.02)); in XLA_TEST_F() 154 ComputeAndCompareTuple(&builder, expected, {}, ErrorSpec(0.01)); in XLA_TEST_F()
|
D | axpy_simple_test.cc | 40 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in TEST_F() 52 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 72 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in TEST_F()
|
D | slice_test.cc | 51 ComputeAndCompareR3<float>(&builder, expected, {}, ErrorSpec(0.000001)); in TEST_F() 64 ComputeAndCompareR3<float>(&builder, expected, {}, ErrorSpec(0.000001)); in TEST_F() 77 ComputeAndCompareR3<float>(&builder, expected, {}, ErrorSpec(0.000001)); in TEST_F() 122 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.000001)); in XLA_TEST_F() 136 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.000001)); in TEST_F() 154 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.000001)); in TEST_F() 166 ComputeAndCompareR4(&builder, *expected, {}, ErrorSpec(0.000001)); in TEST_F() 199 ComputeAndCompareLiteral(&builder, expected_literal, {}, ErrorSpec(0.000001), in XLA_TEST_F() 494 ComputeAndCompareR4(&builder, *expected, {arg.get()}, ErrorSpec(0.000001)); in Run()
|
D | iota_test.cc | 45 ErrorSpec{0.0001}); in TEST_P() 76 ComputeAndCompare(&builder, {}, ErrorSpec{0.0001}); in TEST_P() 103 ComputeAndCompare(&builder, {}, ErrorSpec{0.0001}); in TEST_P() 128 ComputeAndCompare(&builder, {}, ErrorSpec{0.0001}); in TEST_P()
|
D | fmax_fmin_test.cc | 40 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in TEST_F() 62 ErrorSpec(0.0001)); in TEST_F() 84 ErrorSpec(0.0001)); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
D | gpu_kernel_tiling_test.cc | 74 EXPECT_TRUE(RunAndCompareNoHloPasses(kHloString, ErrorSpec{0.0})); in TEST_F() 127 EXPECT_TRUE(RunAndCompareNoHloPasses(kHloString, ErrorSpec{0.0})); in TEST_F() 161 EXPECT_TRUE(RunAndCompareNoHloPasses(kHloString, ErrorSpec{0.0})); in TEST_F() 250 EXPECT_TRUE(RunAndCompareNoHloPasses(kHloString, ErrorSpec{0.0})); in TEST_F() 284 EXPECT_TRUE(RunAndCompareNoHloPasses(kHloString, ErrorSpec{0.0})); in TEST_F() 317 EXPECT_TRUE(RunAndCompareNoHloPasses(kHloString, ErrorSpec{1.0e-5, 1.0e-5})); in TEST_F() 362 EXPECT_TRUE(RunAndCompareNoHloPasses(kHloString, ErrorSpec{1.0e-5, 1.0e-5})); in TEST_F() 412 EXPECT_TRUE(RunAndCompareNoHloPasses(kHloString, ErrorSpec{1.0e-5, 1.0e-5})); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | self_adjoint_eig_test.cc | 154 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 167 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 180 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 192 {a_data.get()}, ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 204 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 220 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 237 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 260 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 273 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 286 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() [all …]
|
D | svd_test.cc | 149 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 161 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 174 ErrorSpec(1e-2, 1e-2)); in XLA_TEST_F() 186 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 203 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 215 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 227 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 239 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 251 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() 263 ErrorSpec(1e-3, 1e-3)); in XLA_TEST_F() [all …]
|
/external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
D | special_math_test.py | 82 ErrorSpec = collections.namedtuple("ErrorSpec", ["rtol", "atol"]) variable 156 _error32 = ErrorSpec(rtol=1e-4, atol=0.) 157 _error64 = ErrorSpec(rtol=1e-6, atol=0.) 230 _error32 = ErrorSpec(rtol=1e-4, atol=0.) 231 _error64 = ErrorSpec(rtol=1e-4, atol=0.) 245 _error32 = ErrorSpec(rtol=0.1, atol=1e-7) 246 _error64 = ErrorSpec(rtol=0.1, atol=1e-7) 259 _error32 = ErrorSpec(rtol=1e-6, atol=1e-14) 260 _error64 = ErrorSpec(rtol=1e-6, atol=1e-14) 266 _error32 = ErrorSpec(rtol=1e-4, atol=0) [all …]
|