Home
last modified time | relevance | path

Searched refs:error_spec_ (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconstants_test.cc41 const ErrorSpec error_spec_{1e-3, 1e-5}; member in xla::__anon141266ec0111::ConstantsTest
48 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in TEST_F()
57 ComputeAndCompareR1<float>(&builder, constant, {}, error_spec_); in TEST_F()
84 ComputeAndCompareR1<float>(&builder, constant, {}, error_spec_); in TEST_F()
94 ComputeAndCompareR1<float>(&builder, constant, {}, error_spec_); in TEST_F()
101 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 2), {}, error_spec_); in TEST_F()
111 ComputeAndCompareR2<float>(&builder, *constant, {}, error_spec_); in TEST_F()
151 ComputeAndCompareR4<float>(&builder, input_array, {}, error_spec_); in TEST_F()
157 ComputeAndCompareR4<float>(&builder, input_array, {}, error_spec_); in TEST_F()
171 LiteralSlice(result, {0}), error_spec_); in TEST_F()
[all …]
Dselect_test.cc33 ErrorSpec error_spec_{0.0001}; member in xla::__anon62562faf0111::SelectTest
43 ComputeAndCompareR0<float>(&builder, 123.0f, {}, error_spec_); in TEST_F()
63 ComputeAndCompareR0<float>(&builder, 42.0f, {}, error_spec_); in TEST_F()
73 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
86 error_spec_); in TEST_F()
100 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
117 error_spec_); in TEST_F()
133 error_spec_); in TEST_F()
153 error_spec_); in TEST_F()
194 error_spec_); in TEST_F()
[all …]
Dconvolution_variants_test.cc48 ErrorSpec error_spec_ = ErrorSpec(1e-1, 1e-5); member in xla::__anon37a439160111::ConvolutionVariantsTest
50 ErrorSpec error_spec_ = ErrorSpec(1e-4, 1e-2);
66 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
81 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
98 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
113 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
128 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
143 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
158 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
173 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
[all …]
Dbroadcast_test.cc50 error_spec_)); in XLA_TEST_F()
67 error_spec_)); in XLA_TEST_F()
90 LiteralSlice(result, {0}), error_spec_)); in XLA_TEST_F()
94 LiteralSlice(result, {1}), error_spec_)); in XLA_TEST_F()
111 error_spec_)); in XLA_TEST_F()
130 error_spec_)); in XLA_TEST_F()
148 result, error_spec_)); in XLA_TEST_F()
170 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
200 LiteralUtil::CreateR4FromArray4D<float>(expected), result, error_spec_)); in TEST_F()
222 result, error_spec_)); in XLA_TEST_F()
[all …]
Dtuple_test.cc41 ErrorSpec error_spec_{0.0001}; member in xla::__anonfbd1e8b90111::TupleTest
60 ComputeAndCompareTuple(&builder, value, {}, error_spec_); in XLA_TEST_F()
74 ComputeAndCompareTuple(&builder, value, {}, error_spec_); in XLA_TEST_F()
95 ComputeAndCompareTuple(&builder, expected, {}, error_spec_); in XLA_TEST_F()
107 ComputeAndCompareTuple(&builder, expected, {}, error_spec_); in XLA_TEST_F()
115 ComputeAndCompareTuple(&builder, expected, {}, error_spec_); in XLA_TEST_F()
131 error_spec_); in XLA_TEST_F()
142 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 101), {}, error_spec_); in XLA_TEST_F()
182 ComputeAndCompareR2<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
202 ComputeAndCompareTuple(&builder, expected, {}, error_spec_); in XLA_TEST_F()
[all …]
Darray_elementwise_ops_test.cc45 ErrorSpec error_spec_{0.0001, 0.0001}; member in xla::__anonc1d9b1400111::ArrayElementwiseOpTest
58 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
67 error_spec_); in XLA_TEST_F()
99 ComputeAndCompareR1<complex64>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
110 {}, error_spec_); in XLA_TEST_F()
215 error_spec_); in XLA_TEST_F()
224 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
237 error_spec_); in XLA_TEST_F()
246 ComputeAndCompareR1<complex64>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
382 error_spec_); in TEST_P()
[all …]
Dconditional_test.cc170 ErrorSpec error_spec_{0.001}; member in xla::__anonf9fe2de30111::ConditionalOpTest
188 ComputeAndCompareR0<float>(&builder, 56.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
217 error_spec_); in XLA_TEST_P()
231 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
269 error_spec_); in XLA_TEST_P()
284 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
297 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
311 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
324 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
338 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
[all …]
Dtranspose_test.cc33 ErrorSpec error_spec_{0.0001}; member in xla::__anon37af8cee0111::TransposeTest
44 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 0), {}, error_spec_); in XLA_TEST_F()
52 ComputeAndCompareR2<float>(&builder, Array2D<float>(42, 0), {}, error_spec_); in XLA_TEST_F()
60 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 7), {}, error_spec_); in XLA_TEST_F()
73 ComputeAndCompareR2<float>(&builder, expected, {}, error_spec_); in TEST_F()
129 ComputeAndCompareR2<float>(&builder, expected, {}, error_spec_); in TEST_F()
Ddot_operation_test.cc43 ErrorSpec error_spec_{0.0001, 1e-5}; member in xla::__anondc2748460111::DotOperationTest
105 this->error_spec_); in XLA_TYPED_TEST()
120 this->error_spec_); in XLA_TYPED_TEST()
131 this->error_spec_); in XLA_TYPED_TEST()
142 this->error_spec_); in XLA_TYPED_TEST()
157 this->error_spec_); in XLA_TYPED_TEST()
169 this->error_spec_); in XLA_TYPED_TEST()
181 this->error_spec_); in XLA_TYPED_TEST()
192 &builder, Array2D<T>(2, 2, static_cast<T>(0.0f)), {}, this->error_spec_); in XLA_TYPED_TEST()
216 this->error_spec_ = ErrorSpec{0.0001, 1e-3}; in XLA_TYPED_TEST()
[all …]
Dscalar_computations_test.cc43 ErrorSpec error_spec_{0.0001}; member in xla::__anon672da6aa0111::ScalarComputationsTest
74 ComputeAndCompareR0<float>(&builder, 2.1f, {}, error_spec_); in XLA_TEST_F()
81 ComputeAndCompareR0<float>(&builder, -2.1f, {}, error_spec_); in XLA_TEST_F()
95 ComputeAndCompareR0<float>(&builder, 7.6f, {}, error_spec_); in XLA_TEST_F()
148 ComputeAndCompareR0<float>(&builder, -3.4f, {}, error_spec_); in XLA_TEST_F()
176 ComputeAndCompareR0<float>(&builder, 5.775f, {}, error_spec_); in XLA_TEST_F()
256 error_spec_); in XLA_TEST_F()
263 ComputeAndCompareR0<float>(&builder, 2.0f, {}, error_spec_); in XLA_TEST_F()
270 ComputeAndCompareR0<float>(&builder, 2.5f, {}, error_spec_); in XLA_TEST_F()
575 ComputeAndCompareR0<float>(&builder, 123.0f, {}, error_spec_); in XLA_TEST_F()
[all …]
Dconvolution_test_1d.cc50 ErrorSpec error_spec_ = ErrorSpec(1e-2, 1e-3); member in xla::__anone9f8da230111::ConvolutionTest
52 ErrorSpec error_spec_ = ErrorSpec(1e-4, 1e-3);
130 error_spec_); in TestImpl()
231 error_spec_); in XLA_TEST_F()
266 error_spec_); in RunTest()
301 error_spec_); in XLA_TEST_F()
332 error_spec_); in XLA_TEST_F()
368 error_spec_); in RunTest()
Dlocal_client_allocation_test.cc36 ErrorSpec error_spec_{0.0001}; member in xla::__anon06b6ff3d0111::LocalClientAllocationTest
61 {2.0f, 4.0f, 6.0f}, ShapedBufferToLiteral(*result), error_spec_); in XLA_TEST_F()
95 {2.0f, 4.0f, 6.0f}, ShapedBufferToLiteral(result), error_spec_); in XLA_TEST_F()
Dconvolution_test.cc50 ErrorSpec error_spec_ = ErrorSpec(1e-2, 1e-3); member in xla::__anon035829af0111::ConvolutionTest
52 ErrorSpec error_spec_ = ErrorSpec(1e-4, 1e-3);
103 ComputeAndCompare(&builder, {}, error_spec_); in RunTest()
135 error_spec_); in RunTest()
169 error_spec_); in RunTest()
204 error_spec_); in RunTest()
236 error_spec_); in RunTest()
295 error_spec_); in XLA_TEST_F()
358 error_spec_); in RunTest()
422 error_spec_); in RunTest()
[all …]
Dvector_ops_simple_test.cc48 ErrorSpec error_spec_{0.0001}; member in xla::__anon11ea53fc0111::VecOpsSimpleTest
61 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
116 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
147 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
162 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
193 error_spec_); in XLA_TEST_F()
233 error_spec_); in XLA_TEST_F()
Dbfloat16_test.cc49 const ErrorSpec error_spec_{0.001, 0.001}; member in xla::__anon9181ea4e0111::Bfloat16Test
59 error_spec_); in XLA_TEST_F()
76 error_spec_); in XLA_TEST_F()
Dcompilation_cache_test.cc52 LiteralUtil::CreateR0<float>(expected_result), result, error_spec_)); in ExecuteComputationR0F32()
68 LiteralUtil::CreateR2<float>(expected_result), result, error_spec_)); in ExecuteComputationR2F32()
72 ErrorSpec error_spec_{0.0001}; member in xla::__anon7f6b70610111::CompilationCacheTest
Dlocal_client_execute_test.cc53 ErrorSpec error_spec_{0.0001}; member in xla::__anone0febaca0111::LocalClientExecuteTest
63 error_spec_); in XLA_TEST_F()
76 error_spec_); in XLA_TEST_F()
89 error_spec_); in XLA_TEST_F()
103 {2.0f, 4.0f, 6.0f}, ShapedBufferToLiteral(result), error_spec_); in XLA_TEST_F()
120 {2.0f, 4.0f, 6.0f}, ShapedBufferToLiteral(result), error_spec_); in XLA_TEST_F()
147 error_spec_); in XLA_TEST_F()
154 error_spec_); in XLA_TEST_F()
180 error_spec_); in XLA_TEST_F()
193 error_spec_); in XLA_TEST_F()
[all …]
Dhalf_test.cc35 const ErrorSpec error_spec_{0.001, 0.001}; member in xla::__anon51ab1f0a0111::HalfTestBase
67 ComputeAndCompareR1<half>(&builder, expected, {x_data.get()}, error_spec_); in XLA_TEST_P()
162 error_spec_); in XLA_TEST_P()
Dcopy_test.cc105 LiteralTestUtil::ExpectR0Near<float>(42.0f, result, error_spec_); in XLA_TEST_F()
126 error_spec_); in XLA_TEST_F()
155 error_spec_); in XLA_TEST_F()
Dmultioutput_fusion_test.cc52 MultiOutputFusionTest() { error_spec_ = ErrorSpec{0.0001, 1e-2}; } in MultiOutputFusionTest()
120 EXPECT_TRUE(LiteralTestUtil::Near(expect, actual, error_spec_)); in RunTest2D()
183 EXPECT_TRUE(LiteralTestUtil::Near(expect, actual, error_spec_)); in RunTest1D()
Dcustom_call_test.cc93 LiteralTestUtil::ExpectR0Near<float>(44.0f, result, error_spec_); in XLA_TEST_F()
114 LiteralTestUtil::ExpectR0Near<float>(10.0f, result, error_spec_); in XLA_TEST_F()
Dbatch_normalization_test.cc90 const ErrorSpec error_spec_{0.001, 0.001}; member in xla::__anon28b54c170111::BatchNormalizationTest
117 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
134 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
145 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
159 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
168 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
229 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling_test.cc62 ErrorSpec error_spec_{0.0001}; member in xla::__anonc9574ad90111::PoolingTest
75 ComputeAndCompareR4<float>(&builder, {{{{5, 4}}}}, {}, error_spec_); in XLA_TEST_F()
88 ComputeAndCompareR4<float>(&builder, {{{{5, 4, 5}}}}, {}, error_spec_); in XLA_TEST_F()
102 {}, error_spec_); in XLA_TEST_F()
118 ComputeAndCompareR4<float>(&builder, {{{{3, 3}}}}, {}, error_spec_); in XLA_TEST_F()
134 ComputeAndCompareR4<float>(&builder, {{{{3, 3, 3}}}}, {}, error_spec_); in XLA_TEST_F()
152 error_spec_); in XLA_TEST_F()
166 ComputeAndCompareR4<float>(&builder, {{{{3, 3}}}}, {}, error_spec_); in XLA_TEST_F()
182 error_spec_); in XLA_TEST_F()
198 error_spec_); in XLA_TEST_F()
[all …]
Dmath_test.cc35 ErrorSpec error_spec_{0.0001}; member in xla::__anon8951088b0111::MathTest
51 {}, error_spec_); in TestLogEdgeCases()
61 error_spec_); in TestLog1pEdgeCases()
109 {}, error_spec_); in TestIsNegZero()
143 ComputeAndCompareR1<T>(&b, expected, {}, error_spec_); in TestSqrtPowInequivalence()
156 ComputeAndCompareR1<T>(&b, expected, {}, error_spec_); in TestErfEdgeCases()
236 ComputeAndCompareR0<float>(&builder, 0.0f, {zero_data.get()}, error_spec_); in XLA_TEST_F()
249 ComputeAndCompareR0<double>(&builder, 0.0f, {zero_data.get()}, error_spec_); in XLA_TEST_F()
282 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
294 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc41 ErrorSpec error_spec_{0.0001, 1e-5}; member in xla::cpu::__anon119164210111::CpuFusionTest
88 LiteralTestUtil::ExpectR1Near<float>({1.0, 40.0, -5.0}, result, error_spec_); in TEST_F()
135 LiteralTestUtil::ExpectR1Near<float>({14.0, 40.0, 40.0}, result, error_spec_); in TEST_F()
222 result, error_spec_); in TEST_F()

12