Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconstants_test.cc41 const ErrorSpec error_spec_{1e-3, 1e-5}; member in xla::__anonccc808100111::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::__anone216c3530111::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::__anone52c5bba0111::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 …]
Dtuple_test.cc41 ErrorSpec error_spec_{0.0001}; member in xla::__anon2e3c97dd0111::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 …]
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 …]
Darray_elementwise_ops_test.cc45 ErrorSpec error_spec_{0.0001, 0.0001}; member in xla::__anon206628640111::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()
191 error_spec_); in XLA_TEST_F()
200 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
213 error_spec_); in XLA_TEST_F()
222 ComputeAndCompareR1<complex64>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
358 error_spec_); in TEST_P()
[all …]
Ddot_operation_test.cc43 ErrorSpec error_spec_{0.0001, 1e-5}; member in xla::__anon4b50db6a0111::DotOperationTest
100 this->error_spec_); in XLA_TYPED_TEST()
115 this->error_spec_); in XLA_TYPED_TEST()
126 this->error_spec_); in XLA_TYPED_TEST()
137 this->error_spec_); in XLA_TYPED_TEST()
152 this->error_spec_); in XLA_TYPED_TEST()
164 this->error_spec_); in XLA_TYPED_TEST()
176 this->error_spec_); in XLA_TYPED_TEST()
187 &builder, Array2D<T>(2, 2, static_cast<T>(0.0f)), {}, this->error_spec_); in XLA_TYPED_TEST()
211 this->error_spec_ = ErrorSpec{0.0001, 1e-3}; in XLA_TYPED_TEST()
[all …]
Dconditional_test.cc170 ErrorSpec error_spec_{0.001}; member in xla::__anonb6a0a8070111::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::__anonf0652e120111::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()
Dscalar_computations_test.cc43 ErrorSpec error_spec_{0.0001}; member in xla::__anon552a04ce0111::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.cc50 ErrorSpec error_spec_ = ErrorSpec(1e-2, 1e-3); member in xla::__anonbffaa3d30111::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()
267 error_spec_); in XLA_TEST_F()
302 error_spec_); in RunTest()
337 error_spec_); in XLA_TEST_F()
[all …]
Dlocal_client_allocation_test.cc36 ErrorSpec error_spec_{0.0001}; member in xla::__anon3a3dcf610111::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()
Dvector_ops_simple_test.cc48 ErrorSpec error_spec_{0.0001}; member in xla::__anon5430d9200111::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::__anon03b5f6f20111::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::__anonc1b1f5850111::CompilationCacheTest
Dlocal_client_execute_test.cc53 ErrorSpec error_spec_{0.0001}; member in xla::__anon8e86dd6e0111::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 …]
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()
Dhalf_test.cc35 const ErrorSpec error_spec_{0.001, 0.001}; member in xla::__anon342a79ae0111::HalfTestBase
67 ComputeAndCompareR1<half>(&builder, expected, {x_data.get()}, error_spec_); in XLA_TEST_P()
162 error_spec_); in XLA_TEST_P()
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::__anon16b1aa3b0111::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()
Dfloor_ceil_test.cc67 const ErrorSpec error_spec_{0.0001}; member in xla::__anonb9096bd10111::FloorCeilTest
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling_test.cc62 ErrorSpec error_spec_{0.0001}; member in xla::__anon1a49097d0111::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::__anon4206a9af0111::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()
269 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
281 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
289 ComputeAndCompareR1<float>(&builder, {0, 0}, {}, 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::__anon41acbfc50111::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