| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | transpose_test.cc | 82 ComputeAndCompareR3<int32>(&builder, Array3D<int32>(2, 3, 0), {}); in XLA_TEST_F()
93 ComputeAndCompareR3<int32>(&builder, expected, {}); in TEST_F()
104 ComputeAndCompareR3<int32>(&builder, expected, {}); in TEST_F()
115 ComputeAndCompareR3<int32>(&builder, expected, {}); in TEST_F()
173 ComputeAndCompareR3<int32>(&builder, expected, {}); in TestTransposeConstant021()
|
| D | constants_test.cc | 119 ComputeAndCompareR3<float>(&builder, Array3D<float>(3, 0, 2), {}); in TEST_F()
134 ComputeAndCompareR3<float>(&builder, array3d, {}); in TEST_F()
|
| D | concat_test.cc | 242 ComputeAndCompareR3<float>(&builder, Array3D<float>(3, 0, 3), {}, in XLA_TEST_F()
269 ComputeAndCompareR3<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
313 ComputeAndCompareR3<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
492 ComputeAndCompareR3<float>(&builder, expected, {p0.get(), p1.get()}); in XLA_TEST_F()
633 ComputeAndCompareR3<float>(&builder, *concat1, in XLA_TEST_F()
|
| D | multidimensional_slice_test.cc | 53 ComputeAndCompareR3<float>(&builder, expected_3d, {}, ErrorSpec(0.000001)); in XLA_TEST_F()
|
| D | log_test.cc | 36 ComputeAndCompareR3<float>(&builder, Array3D<float>(3, 0, 0), {}, in XLA_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()
|
| D | client_library_test_base.h | 161 void ComputeAndCompareR3(XlaBuilder* builder,
165 void ComputeAndCompareR3(XlaBuilder* builder,
495 void ClientLibraryTestBase::ComputeAndCompareR3( in ComputeAndCompareR3() function
505 void ClientLibraryTestBase::ComputeAndCompareR3( in ComputeAndCompareR3() function
|
| D | cholesky_test.cc | 149 ComputeAndCompareR3<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
|
| D | broadcast_simple_test.cc | 203 ComputeAndCompareR3<float>(&b, expected, {}, ErrorSpec(0.0001));
221 ComputeAndCompareR3<float>(&b, expected, {}, ErrorSpec(0.0001));
263 ComputeAndCompareR3<bool>(&b, expected, {x_data.get(), y_data.get()}); in XLA_TEST_F()
|
| D | convolution_test.cc | 265 ComputeAndCompareR3<float>(&builder, expected, in XLA_TEST_F()
300 ComputeAndCompareR3<T>(&builder, expected, in RunTest()
335 ComputeAndCompareR3<float>(&builder, expected, in XLA_TEST_F()
366 ComputeAndCompareR3<float>(&builder, expected, in XLA_TEST_F()
402 ComputeAndCompareR3<T>(&builder, expected, in RunTest()
|
| D | array_elementwise_ops_test.cc | 1965 ComputeAndCompareR3<int32>(&builder, expected, {}); in XLA_TEST_F()
1976 ComputeAndCompareR3<int32>(&builder, expected, {}); in XLA_TEST_F()
2204 ComputeAndCompareR3<float>( in XLA_TEST_F()
2710 ComputeAndCompareR3<float>(&builder, expected_3d, {}, error_spec_);
2734 ComputeAndCompareR3<float>(&builder, expected_3d, {}, error_spec_); in XLA_TEST_F()
2765 ComputeAndCompareR3<float>(&builder, expected_3d, {}, error_spec_); in XLA_TEST_F()
2797 ComputeAndCompareR3<float>(&builder, expected_3d, {}, error_spec_); in XLA_TEST_F()
|
| D | triangular_solve_test.cc | 430 ComputeAndCompareR3<float>(&builder, bvals, {a_data.get(), b_data.get()}, in XLA_TEST_F()
|
| D | pad_test.cc | 465 ComputeAndCompareR3<float>(&b, expected, {}, DefaultErrorSpec()); in XLA_TEST_P()
|
| D | convolution_variants_test.cc | 1311 ComputeAndCompareR3<float>(&builder, {{{10}}}, {}, error_spec_); in XLA_TEST_F()
1330 ComputeAndCompareR3<float>(&builder, {{{13, 24, 130}}}, {}, error_spec_); in XLA_TEST_F()
|
| D | map_test.cc | 409 ComputeAndCompareR3<int32>(&builder, Array3D<int32>(3, 0, 2), in XLA_TEST_F()
|
| D | dynamic_ops_test.cc | 556 ComputeAndCompareR3<T>(&builder, expected_values, in RunR3Contiguous()
|
| D | dot_operation_test.cc | 634 this->template ComputeAndCompareR3<T>( in XLA_TYPED_TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix_test.cc | 48 ComputeAndCompareR3<int32>(&builder, expected, {a_data.get()}); in XLA_TEST_F()
114 ComputeAndCompareR3<float>(&builder, {{{33}}, {{292}}}, in XLA_TEST_F()
135 ComputeAndCompareR3<float>(&builder, {{{33}}, {{292}}}, in XLA_TEST_F()
|
| D | self_adjoint_eig_test.cc | 153 ComputeAndCompareR3<float>(&builder, batch_3d_4x4_, {a_data.get()}, in XLA_TEST_F()
166 ComputeAndCompareR3<float>(&builder, batch_3d_4x4_, {a_data.get()}, in XLA_TEST_F()
179 ComputeAndCompareR3<float>(&builder, batch_3d_4x4_, {a_data.get()}, in XLA_TEST_F()
191 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(batch_3d_4x4_), in XLA_TEST_F()
|
| D | svd_test.cc | 160 ComputeAndCompareR3<float>(&builder, batch_3d_4x5_, {a_data.get()}, in XLA_TEST_F()
173 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(2, 4), {a_data.get()}, in XLA_TEST_F()
185 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(2, 5), {a_data.get()}, in XLA_TEST_F()
|
| D | qr_test.cc | 89 ComputeAndCompareR3<float>(&builder, a_vals, {a_data.get()}, in XLA_TEST_F()
|
| D | slicing_test.cc | 82 ComputeAndCompareR3<float>(&builder, {{{3, 6, 0, 1}}, {{24, 61, 82, 48}}}, in XLA_TEST_F()
|