| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing_test.cc | 68 ComputeAndCompareR2<float>(&builder, {{10}}, in XLA_TEST_F()
102 ComputeAndCompareR2<float>( in XLA_TEST_F()
117 ComputeAndCompareR2<int>(&builder, {{1, 1}, {4, 3}}, in XLA_TEST_F()
131 ComputeAndCompareR2<int>(&builder, {{1, 1}, {4, 3}}, in XLA_TEST_F()
148 ComputeAndCompareR2<int>( in XLA_TEST_F()
166 ComputeAndCompareR2<float>( in XLA_TEST_F()
186 ComputeAndCompareR2<float>( in XLA_TEST_F()
199 ComputeAndCompareR2<float>(&builder, {{}, {}, {}}, in XLA_TEST_F()
230 ComputeAndCompareR2<float>(&builder, in XLA_TEST_F()
|
| D | quantize_test.cc | 222 ComputeAndCompareR2<bfloat16>(&builder, expected, {}); in XLA_TEST_F()
245 ComputeAndCompareR2<bfloat16>(&builder, expected, {}); in XLA_TEST_F()
275 ComputeAndCompareR2<bfloat16>(&builder, expected, {}); in XLA_TEST_F()
305 ComputeAndCompareR2<bfloat16>(&builder, expected, {}); in XLA_TEST_F()
319 ComputeAndCompareR2<bfloat16>(&builder, expected, {}); in XLA_TEST_F()
333 ComputeAndCompareR2<bfloat16>(&builder, expected, {}); in XLA_TEST_F()
|
| D | qr_test.cc | 62 ComputeAndCompareR2<float>(&builder, a_vals, {a_data.get()}, in XLA_TEST_F()
91 ComputeAndCompareR2<float>(&builder, a_vals, {a_data.get()}, in XLA_TEST_F()
|
| D | matrix_test.cc | 86 ComputeAndCompareR2<T>(&builder, kv.second, {a_data.get()}); in TestMatrixDiagonal()
106 ComputeAndCompareR2<T>(&builder, kv.second, {a_data.get(), new_diag.get()}); in TestSetMatrixDiagonal()
|
| D | svd_test.cc | 133 ComputeAndCompareR2<float>(&builder, simple_2d_4x4_, {a_data.get()}, in XLA_TEST_F()
187 ComputeAndCompareR2<float>(&builder, singular_values, {a_data.get()}, in XLA_TEST_F()
|
| D | sorting_test.cc | 131 ComputeAndCompareR2<float>(&builder, expected_array, {}, error_spec); in XLA_TEST_F()
|
| D | constants_test.cc | 80 ComputeAndCompareR2<int32>(&builder, {{0, 0}, {0, 0}}, {}); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | transpose_test.cc | 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()
142 ComputeAndCompareR2<float>(&builder, *expected, {}, ErrorSpec(1e-4)); in TEST_F()
154 ComputeAndCompareR2<float>(&builder, *expected, {}, ErrorSpec(1e-4)); in TEST_F()
|
| D | triangular_solve_test.cc | 116 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 10), in XLA_TEST_F()
137 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
158 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
179 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
200 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
222 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
244 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
263 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
285 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
307 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}, in XLA_TEST_F()
[all …]
|
| D | binop_scaling_test.cc | 44 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001)); in TEST_F()
60 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001)); in TEST_F()
76 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001)); in TEST_F()
92 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001)); in TEST_F()
109 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001)); in TEST_F()
|
| D | dot_operation_test.cc | 156 this->template ComputeAndCompareR2<T>(&builder, Array2D<T>(0, 0), {}, in XLA_TYPED_TEST()
168 this->template ComputeAndCompareR2<T>(&builder, Array2D<T>(0, 3), {}, in XLA_TYPED_TEST()
180 this->template ComputeAndCompareR2<T>(&builder, Array2D<T>(3, 0), {}, in XLA_TYPED_TEST()
191 this->template ComputeAndCompareR2<T>( in XLA_TYPED_TEST()
219 this->template ComputeAndCompareR2<T>( in XLA_TYPED_TEST()
248 ComputeAndCompareR2<T>(&builder, expected, in TestImpl()
301 ComputeAndCompareR2(builder, expected, arguments, error_spec); in ComputeAndCompareR2WithError()
309 ComputeAndCompareR2(builder, expected, arguments, error_spec); in ComputeAndCompareR2WithError()
316 ComputeAndCompareR2(builder, expected, arguments); in ComputeAndCompareR2WithError()
535 ComputeAndCompareR2<T>(&builder, expected, in TestImpl()
[all …]
|
| D | select_and_scatter_test.cc | 268 ComputeAndCompareR2<int32>(&builder_, expected, {}); in XLA_TEST_F()
283 ComputeAndCompareR2<float>(&builder_, expected, {}, ErrorSpec(1e-7)); in XLA_TEST_F()
297 ComputeAndCompareR2<int32>(&builder_, expected, {}); in XLA_TEST_F()
309 ComputeAndCompareR2<int32>(&builder_, expected, {}); in XLA_TEST_F()
321 ComputeAndCompareR2<int32>(&builder_, expected, {}); in XLA_TEST_F()
335 ComputeAndCompareR2<int32>(&builder_, expected, {}); in XLA_TEST_F()
348 ComputeAndCompareR2<float>(&builder_, expected, {}, ErrorSpec(1e-7)); in XLA_TEST_F()
|
| D | concat_test.cc | 150 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 0), {}, in XLA_TEST_F()
167 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
181 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
191 ComputeAndCompareR2<float>(&builder, *b_array, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
206 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
216 ComputeAndCompareR2<float>(&builder, *a_array, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
237 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
361 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
382 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
408 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
[all …]
|
| D | broadcast_simple_test.cc | 118 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
130 ComputeAndCompareR2<float>(&b, expected, {param_data.get()}, in XLA_TEST_F()
138 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
145 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
159 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001));
172 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001));
185 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001));
236 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001));
271 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
279 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001));
|
| D | slice_test.cc | 85 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 0), {}); in XLA_TEST_F()
93 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 5), {}); in XLA_TEST_F()
101 ComputeAndCompareR2<float>(&builder, Array2D<float>(2, 0), {}); in XLA_TEST_F()
122 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.000001)); in XLA_TEST_F()
136 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.000001)); in TEST_F()
154 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.000001)); in TEST_F()
416 ComputeAndCompareR2<int32>(&builder, *expected, {arg.get()}); in XLA_TEST_P()
|
| D | vector_ops_reduce_test.cc | 141 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_); in TEST_F()
153 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_); in TEST_F()
164 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_); in TEST_F()
|
| D | cholesky_test.cc | 60 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
124 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
150 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
173 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
|
| D | array_elementwise_ops_test.cc | 813 ComputeAndCompareR2<bool>(&builder, expected_array, {}); in XLA_TEST_F()
841 ComputeAndCompareR2<int32>(&builder, expected_array, {}); in XLA_TEST_F()
869 ComputeAndCompareR2<uint32>(&builder, expected_array, {}); in XLA_TEST_F()
897 ComputeAndCompareR2<bool>(&builder, expected_array, {}); in XLA_TEST_F()
925 ComputeAndCompareR2<int32>(&builder, expected_array, {}); in XLA_TEST_F()
953 ComputeAndCompareR2<uint32>(&builder, expected_array, {}); in XLA_TEST_F()
981 ComputeAndCompareR2<bool>(&builder, expected_array, {}); in XLA_TEST_F()
1009 ComputeAndCompareR2<int32>(&builder, expected_array, {}); in XLA_TEST_F()
1037 ComputeAndCompareR2<uint32>(&builder, expected_array, {}); in XLA_TEST_F()
1062 ComputeAndCompareR2<bool>(&builder, expected_array, {}); in XLA_TEST_F()
[all …]
|
| D | pad_test.cc | 337 ComputeAndCompareR2<float>(&b, *expected, {}, DefaultErrorSpec()); in XLA_TEST_P()
359 ComputeAndCompareR2<float>(&b, *expected, {}, DefaultErrorSpec()); in XLA_TEST_P()
386 ComputeAndCompareR2<float>(&b, *expected, {}, DefaultErrorSpec()); in XLA_TEST_P()
414 ComputeAndCompareR2<float>(&b, *expected, {}, DefaultErrorSpec()); in XLA_TEST_P()
442 ComputeAndCompareR2<float>(&b, *expected, {}, DefaultErrorSpec()); in XLA_TEST_P()
|
| D | multidimensional_slice_test.cc | 42 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.000001)); in XLA_TEST_F()
|
| D | constants_test.cc | 101 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 2), {}, error_spec_); in TEST_F()
111 ComputeAndCompareR2<float>(&builder, *constant, {}, error_spec_); in TEST_F()
|
| D | matrix_ops_simple_test.cc | 154 ComputeAndCompareR2<T>(&builder, expected, {}, error_spec); in TestImpl()
235 ComputeAndCompareR2<T>(&builder, expected, in TestImpl()
|
| D | tuple_test.cc | 130 ComputeAndCompareR2<float>(&builder, Array2D<float>(constant_matrix), {}, 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()
275 ComputeAndCompareR2<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
|
| D | params_test.cc | 102 ComputeAndCompareR2<float>(&builder, Array2D<float>(3, 0), in XLA_TEST_F()
117 ComputeAndCompareR2<float>(&builder, expected_array, {param0_data.get()}, in XLA_TEST_F()
509 ComputeAndCompareR2(&builder, expected, {data.get()}, ErrorSpec(1e-3)); in XLA_TEST_F()
|
| D | client_library_test_base.h | 153 void ComputeAndCompareR2(XlaBuilder* builder,
157 void ComputeAndCompareR2(XlaBuilder* builder,
499 void ClientLibraryTestBase::ComputeAndCompareR2( in ComputeAndCompareR2() function
509 void ClientLibraryTestBase::ComputeAndCompareR2( in ComputeAndCompareR2() function
|