/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | constants_test.cc | 41 const ErrorSpec error_spec_{1e-3, 1e-5}; member in xla::__anona447d60e0111::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 …]
|
D | convolution_variants_test.cc | 48 ErrorSpec error_spec_ = ErrorSpec(1e-1, 1e-5); member in xla::__anon2288aef80111::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 …]
|
D | select_test.cc | 33 ErrorSpec error_spec_{0.0001}; member in xla::__anon512322110111::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 …]
|
D | array_elementwise_ops_test.cc | 45 ErrorSpec error_spec_{0.0001, 0.0001}; member in xla::__anon094ce3620111::ArrayElementwiseOpTest 57 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F() 66 error_spec_); in XLA_TEST_F() 90 ComputeAndCompareR1<complex64>(&builder, {}, {}, error_spec_); in XLA_TEST_F() 101 {}, error_spec_); in XLA_TEST_F() 182 error_spec_); in XLA_TEST_F() 191 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F() 204 error_spec_); in XLA_TEST_F() 213 ComputeAndCompareR1<complex64>(&builder, {}, {}, error_spec_); in XLA_TEST_F() 349 error_spec_); in TEST_P() [all …]
|
D | tuple_test.cc | 41 ErrorSpec error_spec_{0.0001}; member in xla::__anonadb916db0111::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 …]
|
D | broadcast_test.cc | 50 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 …]
|
D | dot_operation_test.cc | 41 ErrorSpec error_spec_{0.0001, 1e-5}; member in xla::__anon59e378680111::DotOperationTest 98 this->error_spec_); in XLA_TYPED_TEST() 113 this->error_spec_); in XLA_TYPED_TEST() 124 this->error_spec_); in XLA_TYPED_TEST() 135 this->error_spec_); in XLA_TYPED_TEST() 150 this->error_spec_); in XLA_TYPED_TEST() 162 this->error_spec_); in XLA_TYPED_TEST() 174 this->error_spec_); in XLA_TYPED_TEST() 185 &builder, Array2D<T>(2, 2, static_cast<T>(0.0f)), {}, this->error_spec_); in XLA_TYPED_TEST() 209 this->error_spec_ = ErrorSpec{0.0001, 1e-3}; in XLA_TYPED_TEST() [all …]
|
D | conditional_test.cc | 170 ErrorSpec error_spec_{0.001}; member in xla::__anon6d4bed850111::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 …]
|
D | scalar_computations_test.cc | 43 ErrorSpec error_spec_{0.0001}; member in xla::__anon6689e84c0111::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() 246 error_spec_); in XLA_TEST_F() 253 ComputeAndCompareR0<float>(&builder, 2.0f, {}, error_spec_); in XLA_TEST_F() 260 ComputeAndCompareR0<float>(&builder, 2.5f, {}, error_spec_); in XLA_TEST_F() 565 ComputeAndCompareR0<float>(&builder, 123.0f, {}, error_spec_); in XLA_TEST_F() [all …]
|
D | transpose_test.cc | 33 ErrorSpec error_spec_{0.0001}; member in xla::__anonc7e4fc100111::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()
|
D | convolution_test.cc | 50 ErrorSpec error_spec_ = ErrorSpec(1e-2, 1e-4); member in xla::__anon76a5e9510111::ConvolutionTest 52 ErrorSpec error_spec_ = ErrorSpec(1e-4, 1e-4); 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 …]
|
D | vector_ops_simple_test.cc | 48 ErrorSpec error_spec_{0.0001}; member in xla::__anon74a5c51e0111::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()
|
D | local_client_allocation_test.cc | 36 ErrorSpec error_spec_{0.0001}; member in xla::__anonf7bf41df0111::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()
|
D | bfloat16_test.cc | 49 const ErrorSpec error_spec_{0.001, 0.001}; member in xla::__anon6754fd300111::Bfloat16Test 59 error_spec_); in XLA_TEST_F() 76 error_spec_); in XLA_TEST_F()
|
D | local_client_execute_test.cc | 52 ErrorSpec error_spec_{0.0001}; member in xla::__anoncbe330ac0111::LocalClientExecuteTest 62 error_spec_); in XLA_TEST_F() 75 error_spec_); in XLA_TEST_F() 88 error_spec_); in XLA_TEST_F() 102 {2.0f, 4.0f, 6.0f}, ShapedBufferToLiteral(result), error_spec_); in XLA_TEST_F() 119 {2.0f, 4.0f, 6.0f}, ShapedBufferToLiteral(result), error_spec_); in XLA_TEST_F() 146 error_spec_); in XLA_TEST_F() 153 error_spec_); in XLA_TEST_F() 178 error_spec_); in XLA_TEST_F() 190 error_spec_); in XLA_TEST_F() [all …]
|
D | compilation_cache_test.cc | 52 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::__anone226e1830111::CompilationCacheTest
|
D | copy_test.cc | 105 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()
|
D | half_test.cc | 35 const ErrorSpec error_spec_{0.001, 0.001}; member in xla::__anonea7410ec0111::HalfTestBase 67 ComputeAndCompareR1<half>(&builder, expected, {x_data.get()}, error_spec_); in XLA_TEST_P() 162 error_spec_); in XLA_TEST_P()
|
D | batch_normalization_test.cc | 89 const ErrorSpec error_spec_{0.001, 0.001}; member in xla::__anon28118db90111::BatchNormalizationTest 116 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P() 133 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P() 144 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P() 158 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P() 167 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P() 228 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
|
D | custom_call_test.cc | 93 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()
|
D | multioutput_fusion_test.cc | 52 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()
|
D | floor_ceil_test.cc | 67 const ErrorSpec error_spec_{0.0001}; member in xla::__anon8082f98f0111::FloorCeilTest
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | pooling_test.cc | 62 ErrorSpec error_spec_{0.0001}; member in xla::__anona65cfebb0111::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 …]
|
D | math_test.cc | 32 ErrorSpec error_spec_{0.0001}; member in xla::__anon198677ad0111::MathTest 48 {}, error_spec_); in TestLogEdgeCases() 58 error_spec_); in TestLog1pEdgeCases() 106 {}, error_spec_); in TestIsNegZero() 136 ComputeAndCompareR1<T>(&b, expected, {}, error_spec_); in TestSqrtPowInequivalence() 149 ComputeAndCompareR1<T>(&b, expected, {}, error_spec_); in TestErfEdgeCases() 224 ComputeAndCompareR0<float>(&builder, 0.0f, {zero_data.get()}, error_spec_); in XLA_TEST_F() 235 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F() 247 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F() 255 ComputeAndCompareR1<float>(&builder, {0, 0}, {}, error_spec_); in XLA_TEST_F() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
D | cpu_fusion_test.cc | 41 ErrorSpec error_spec_{0.0001, 1e-5}; member in xla::cpu::__anon476ada830111::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()
|