Home
last modified time | relevance | path

Searched refs:ComputeAndCompareR3 (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dtranspose_test.cc82 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()
Dconvolution_test_1d.cc229 ComputeAndCompareR3<float>(&builder, expected, in XLA_TEST_F()
264 ComputeAndCompareR3<T>(&builder, expected, in RunTest()
299 ComputeAndCompareR3<float>(&builder, expected, in XLA_TEST_F()
330 ComputeAndCompareR3<float>(&builder, expected, in XLA_TEST_F()
366 ComputeAndCompareR3<T>(&builder, expected, in RunTest()
Dmultidimensional_slice_test.cc53 ComputeAndCompareR3<float>(&builder, expected_3d, {}, ErrorSpec(0.000001)); in XLA_TEST_F()
Dconstants_test.cc119 ComputeAndCompareR3<float>(&builder, Array3D<float>(3, 0, 2), {}); in TEST_F()
134 ComputeAndCompareR3<float>(&builder, array3d, {}); in TEST_F()
Dlog_test.cc36 ComputeAndCompareR3<float>(&builder, Array3D<float>(3, 0, 0), {}, in XLA_TEST_F()
Dconcat_test.cc245 ComputeAndCompareR3<float>(&builder, Array3D<float>(3, 0, 3), {}, in XLA_TEST_F()
272 ComputeAndCompareR3<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
316 ComputeAndCompareR3<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
495 ComputeAndCompareR3<float>(&builder, expected, {p0.get(), p1.get()}); in XLA_TEST_F()
879 ComputeAndCompareR3<float>(&builder, *concat1, in XLA_TEST_F()
Dcholesky_test.cc98 ComputeAndCompareR3<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
212 ComputeAndCompareR3<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
Dslice_test.cc51 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()
Dclient_library_test_base.h163 void ComputeAndCompareR3(XlaBuilder* builder,
167 void ComputeAndCompareR3(XlaBuilder* builder,
526 void ClientLibraryTestBase::ComputeAndCompareR3( in ComputeAndCompareR3() function
536 void ClientLibraryTestBase::ComputeAndCompareR3( in ComputeAndCompareR3() function
Dbroadcast_simple_test.cc203 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()
Darray_elementwise_ops_test.cc2071 ComputeAndCompareR3<int32>(&builder, expected, {}); in XLA_TEST_F()
2082 ComputeAndCompareR3<int32>(&builder, expected, {}); in XLA_TEST_F()
2310 ComputeAndCompareR3<float>( in XLA_TEST_F()
2816 ComputeAndCompareR3<float>(&builder, expected_3d, {}, error_spec_);
2840 ComputeAndCompareR3<float>(&builder, expected_3d, {}, error_spec_); in XLA_TEST_F()
2871 ComputeAndCompareR3<float>(&builder, expected_3d, {}, error_spec_); in XLA_TEST_F()
2903 ComputeAndCompareR3<float>(&builder, expected_3d, {}, error_spec_); in XLA_TEST_F()
Dtriangular_solve_test.cc440 ComputeAndCompareR3<float>(&builder, bvals, {a_data.get(), b_data.get()}, in XLA_TEST_F()
Dpad_test.cc465 ComputeAndCompareR3<float>(&b, expected, {}, DefaultErrorSpec()); in XLA_TEST_P()
Dmap_test.cc409 ComputeAndCompareR3<int32>(&builder, Array3D<int32>(3, 0, 2), in XLA_TEST_F()
Dconvolution_variants_test.cc1311 ComputeAndCompareR3<float>(&builder, {{{10}}}, {}, error_spec_); in XLA_TEST_F()
1330 ComputeAndCompareR3<float>(&builder, {{{13, 24, 130}}}, {}, error_spec_); in XLA_TEST_F()
Ddynamic_ops_test.cc557 ComputeAndCompareR3<T>(&builder, expected_values, in RunR3Contiguous()
Ddot_operation_test.cc678 this->template ComputeAndCompareR3<T>( in XLA_TYPED_TEST()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix_test.cc73 ComputeAndCompareR3<int32>(&builder, expected, {a_data.get()}); in XLA_TEST_F()
147 ComputeAndCompareR3<T>(&builder, kv.second, {a_data.get()}); in TestMatrixDiagonal4D()
194 ComputeAndCompareR3<float>(&builder, {{{33}}, {{292}}}, in XLA_TEST_F()
215 ComputeAndCompareR3<float>(&builder, {{{33}}, {{292}}}, in XLA_TEST_F()
Dself_adjoint_eig_test.cc153 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()
Dsvd_test.cc145 ComputeAndCompareR3<float>(&builder, batch_3d_4x5_, {a_data.get()}, in XLA_TEST_F()
158 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(2, 4), {a_data.get()}, in XLA_TEST_F()
170 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(2, 5), {a_data.get()}, in XLA_TEST_F()
Dslicing_test.cc84 ComputeAndCompareR3<float>(&builder, {{{3, 6, 0, 1}}, {{24, 61, 82, 48}}}, in XLA_TEST_F()
247 ComputeAndCompareR3<int>( in XLA_TEST_F()
Dmath_test.cc442 ComputeAndCompareR3<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
479 ComputeAndCompareR3<half>(&builder, expected, {}, ErrorSpec{1e-3}); in XLA_TEST_F()
504 ComputeAndCompareR3<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
527 ComputeAndCompareR3<half>(&builder, expected, {}, ErrorSpec{1e-4}); in XLA_TEST_F()
Dqr_test.cc123 ComputeAndCompareR3<float>(&builder, a_vals, {a_data.get()}, in XLA_TEST_F()