/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | constants_test.cc | 38 const ErrorSpec error_spec_{1e-3, 1e-5}; member in xla::__anon0610de020111::ConstantsTest 45 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in TEST_F() 54 ComputeAndCompareR1<float>(&builder, constant, {}, error_spec_); in TEST_F() 81 ComputeAndCompareR1<float>(&builder, constant, {}, error_spec_); in TEST_F() 91 ComputeAndCompareR1<float>(&builder, constant, {}, error_spec_); in TEST_F() 98 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 2), {}, error_spec_); in TEST_F() 108 ComputeAndCompareR2<float>(&builder, *constant, {}, error_spec_); in TEST_F() 150 ComputeAndCompareR4<float>(&builder, input_array, {}, error_spec_); in TEST_F() 156 ComputeAndCompareR4<float>(&builder, input_array, {}, error_spec_); in TEST_F() 170 {{1.0}, {2.0}}, LiteralView::Create(*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::__anon8434566c0111::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 | 34 ErrorSpec error_spec_{0.0001}; member in xla::__anon25440e850111::SelectTest 44 ComputeAndCompareR0<float>(&builder, 123.0f, {}, error_spec_); in TEST_F() 64 ComputeAndCompareR0<float>(&builder, 42.0f, {}, error_spec_); in TEST_F() 74 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F() 85 error_spec_); in TEST_F() 99 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F() 114 error_spec_); in TEST_F() 128 error_spec_); in TEST_F() 148 error_spec_); in TEST_F() 189 error_spec_); in TEST_F() [all …]
|
D | broadcast_test.cc | 50 error_spec_); in XLA_TEST_F() 67 error_spec_); in XLA_TEST_F() 90 LiteralView::Create(*result, {0}), error_spec_); in XLA_TEST_F() 94 LiteralView::Create(*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 *result, error_spec_); in TEST_F() 200 *result, error_spec_); in TEST_F() 222 error_spec_); in XLA_TEST_F() [all …]
|
D | tuple_test.cc | 38 ErrorSpec error_spec_{0.0001}; member in xla::__anonea7444cf0111::TupleTest 57 ComputeAndCompareTuple(&builder, *value, {}, error_spec_); in XLA_TEST_F() 71 ComputeAndCompareTuple(&builder, *value, {}, error_spec_); in XLA_TEST_F() 92 ComputeAndCompareTuple(&builder, *expected, {}, error_spec_); in XLA_TEST_F() 104 ComputeAndCompareTuple(&builder, *expected, {}, error_spec_); in XLA_TEST_F() 112 ComputeAndCompareTuple(&builder, *expected, {}, error_spec_); in XLA_TEST_F() 127 error_spec_); in XLA_TEST_F() 137 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 101), {}, error_spec_); in XLA_TEST_F() 175 ComputeAndCompareR2<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F() 194 ComputeAndCompareTuple(&builder, *expected, {}, error_spec_); in XLA_TEST_F() [all …]
|
D | conditional_test.cc | 168 ErrorSpec error_spec_{0.001}; member in xla::__anon6576c2790111::ConditionalOpTest 181 ComputeAndCompareR0<float>(&builder, 56.0f, {}, error_spec_); in XLA_TEST_F() 194 ComputeAndCompareR0<float>(&builder, 12.0f, {}, error_spec_); in XLA_TEST_F() 207 ComputeAndCompareR0<float>(&builder, 12.0f, {}, error_spec_); in XLA_TEST_F() 219 ComputeAndCompareR0<float>(&builder, 12.0f, {}, error_spec_); in XLA_TEST_F() 232 ComputeAndCompareR0<float>(&builder, 12.0f, {}, error_spec_); in XLA_TEST_F() 244 ComputeAndCompareR0<float>(&builder, 12.0f, {}, error_spec_); in XLA_TEST_F() 257 ComputeAndCompareR0<float>(&builder, 12.0f, {}, error_spec_); in XLA_TEST_F() 278 ComputeAndCompareR0<float>(&builder, 12.0f, {}, error_spec_); in XLA_TEST_F() 293 ComputeAndCompareR0<float>(&builder, 68.0f, {}, error_spec_); in XLA_TEST_F() [all …]
|
D | array_elementwise_ops_test.cc | 45 ErrorSpec error_spec_{0.0001, 0.0001}; member in xla::__anona06d79560111::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() 155 error_spec_); in XLA_TEST_F() 164 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F() 177 error_spec_); in XLA_TEST_F() 186 ComputeAndCompareR1<complex64>(&builder, {}, {}, error_spec_); in XLA_TEST_F() 226 error_spec_); in TEST_P() [all …]
|
D | dot_operation_test.cc | 41 ErrorSpec error_spec_{0.0001, 1e-5}; member in xla::__anon08175a5c0111::DotOperationTest 62 ComputeAndCompareR0<float>(&builder, 0.0, {}, error_spec_); in XLA_TEST_F() 71 ComputeAndCompareR1<float>(&builder, {25.0}, {}, error_spec_); in XLA_TEST_F() 81 ComputeAndCompareR0<Element>(&builder, 6.0, {}, error_spec_); in TestOneElementVectorDot() 99 ComputeAndCompareR0<Element>(&builder, 29.5, {}, error_spec_); in TestVectorDot() 120 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 0), {}, error_spec_); in XLA_TEST_F() 129 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 3), {}, error_spec_); in XLA_TEST_F() 139 ComputeAndCompareR2<float>(&builder, Array2D<float>(3, 0), {}, error_spec_); in XLA_TEST_F() 149 error_spec_); in XLA_TEST_F() 170 {lhs_handle.get(), rhs_handle.get()}, error_spec_); in XLA_TEST_F() [all …]
|
D | convolution_test.cc | 50 ErrorSpec error_spec_ = ErrorSpec(1e-2); member in xla::__anon6ed0be450111::ConvolutionTest 52 ErrorSpec error_spec_ = ErrorSpec(1e-4); 111 ComputeAndCompare(&builder, conv, {}, error_spec_); in RunTest() 143 error_spec_); in RunTest() 177 error_spec_); in RunTest() 212 error_spec_); in RunTest() 244 error_spec_); in RunTest() 275 error_spec_); in XLA_TEST_F() 310 error_spec_); in RunTest() 345 error_spec_); in XLA_TEST_F() [all …]
|
D | vector_ops_simple_test.cc | 48 ErrorSpec error_spec_{0.0001}; member in xla::__anonfd2181120111::VecOpsSimpleTest 61 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F() 116 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F() 146 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F() 158 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F() 166 ComputeAndCompareR1<float>(&builder, {0, 0}, {}, error_spec_); in XLA_TEST_F() 175 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F() 187 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F() 202 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F() 233 error_spec_); in XLA_TEST_F() [all …]
|
D | transpose_test.cc | 34 ErrorSpec error_spec_{0.0001}; member in xla::__anon29ae04040111::TransposeTest 45 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 0), {}, error_spec_); in XLA_TEST_F() 53 ComputeAndCompareR2<float>(&builder, Array2D<float>(42, 0), {}, error_spec_); in XLA_TEST_F() 61 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 7), {}, error_spec_); in XLA_TEST_F() 73 ComputeAndCompareR2<float>(&builder, expected, {}, error_spec_); in TEST_F() 125 ComputeAndCompareR2<float>(&builder, expected, {}, error_spec_); in TEST_F()
|
D | scalar_computations_test.cc | 41 ErrorSpec error_spec_{0.0001}; member in xla::__anonfce471400111::ScalarComputationsTest 76 ComputeAndCompareR0<float>(&builder, 2.1f, {}, error_spec_); in XLA_TEST_F() 83 ComputeAndCompareR0<float>(&builder, -2.1f, {}, error_spec_); in XLA_TEST_F() 97 ComputeAndCompareR0<float>(&builder, 7.6f, {}, error_spec_); in XLA_TEST_F() 151 ComputeAndCompareR0<float>(&builder, -3.4f, {}, error_spec_); in XLA_TEST_F() 180 ComputeAndCompareR0<float>(&builder, 5.775f, {}, error_spec_); in XLA_TEST_F() 251 error_spec_); in XLA_TEST_F() 258 ComputeAndCompareR0<float>(&builder, 2.0f, {}, error_spec_); in XLA_TEST_F() 265 ComputeAndCompareR0<float>(&builder, 2.5f, {}, error_spec_); in XLA_TEST_F() 568 ComputeAndCompareR0<float>(&builder, 123.0f, {}, error_spec_); in XLA_TEST_F() [all …]
|
D | bfloat16_test.cc | 51 const ErrorSpec error_spec_{0.001, 0.001}; member in xla::__anonc762d6a40111::Bfloat16Test 61 error_spec_); in XLA_TEST_F() 70 error_spec_); in XLA_TEST_F() 78 error_spec_); in XLA_TEST_F()
|
D | local_client_allocation_test.cc | 36 ErrorSpec error_spec_{0.0001}; member in xla::__anond95d24d30111::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 | local_client_execute_test.cc | 55 ErrorSpec error_spec_{0.0001}; member in xla::__anon2d8ed8200111::LocalClientExecuteTest 66 error_spec_); in XLA_TEST_F() 80 error_spec_); in XLA_TEST_F() 94 error_spec_); in XLA_TEST_F() 109 {2.0f, 4.0f, 6.0f}, *ShapedBufferToLiteral(*result), error_spec_); in XLA_TEST_F() 127 {2.0f, 4.0f, 6.0f}, *ShapedBufferToLiteral(*result), error_spec_); in XLA_TEST_F() 154 error_spec_); in XLA_TEST_F() 161 *ShapedBufferToLiteral(*result_param_swap), error_spec_); in XLA_TEST_F() 186 error_spec_); in XLA_TEST_F() 198 error_spec_); in XLA_TEST_F() [all …]
|
D | compilation_cache_test.cc | 53 *result, error_spec_); in ExecuteComputationR0F32() 70 *result, error_spec_); in ExecuteComputationR2F32() 74 ErrorSpec error_spec_{0.0001}; member in xla::__anon6aa29d770111::CompilationCacheTest
|
D | multioutput_fusion_test.cc | 54 MultiOutputFusionTest() { error_spec_ = ErrorSpec{0.0001, 1e-2}; } in MultiOutputFusionTest() 111 LiteralTestUtil::ExpectNear(expect, *actual, error_spec_); in RunTest2D() 171 LiteralTestUtil::ExpectNear(expect, *actual, error_spec_); in RunTest1D()
|
D | exhaustive_f32_elementwise_op_test.cc | 27 ErrorSpec error_spec_{0.0001, 0.0001, /*relaxed_nans=*/true}; member in xla::__anon41e403740111::ExhaustiveF32ElementwiseOpTest 66 error_spec_); in ExhaustivelyTestF32Op()
|
D | custom_call_test.cc | 83 LiteralTestUtil::ExpectR0Near<float>(44.0f, *result, error_spec_); in XLA_TEST_F() 104 LiteralTestUtil::ExpectR0Near<float>(10.0f, *result, error_spec_); in XLA_TEST_F()
|
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 | batch_normalization_test.cc | 79 const ErrorSpec error_spec_{0.001, 0.001}; member in xla::__anonbe6c16ad0111::BatchNormalizationTest 106 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P() 123 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P() 135 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P() 150 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P() 159 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P() 217 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
|
D | half_test.cc | 36 const ErrorSpec error_spec_{0.001, 0.001}; member in xla::__anondcc750600111::HalfTestBase 68 ComputeAndCompareR1<half>(&builder, expected, {x_data.get()}, error_spec_); in XLA_TEST_P() 175 error_spec_); in XLA_TEST_P()
|
D | floor_ceil_test.cc | 68 const ErrorSpec error_spec_{0.0001}; member in xla::__anon1b6d00030111::FloorCeilTest
|
D | hlo_test_base.h | 213 ErrorSpec error_spec_{0.0001};
|