| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| slicing_test.cc | 66 ComputeAndCompareR2<float>(&builder, {{10}},
100 ComputeAndCompareR2<float>(
115 ComputeAndCompareR2<int>(&builder, {{1, 1}, {4, 3}},
132 ComputeAndCompareR2<float>(
151 ComputeAndCompareR2<float>(
|
| qr_test.cc | 59 ComputeAndCompareR2<float>(&builder, a_vals, {a_data.get()},
|
| quantize_test.cc | 222 ComputeAndCompareR2<bfloat16>(&builder, expected, {});
245 ComputeAndCompareR2<bfloat16>(&builder, expected, {});
275 ComputeAndCompareR2<bfloat16>(&builder, expected, {});
305 ComputeAndCompareR2<bfloat16>(&builder, expected, {});
319 ComputeAndCompareR2<bfloat16>(&builder, expected, {});
333 ComputeAndCompareR2<bfloat16>(&builder, expected, {});
|
| constants_test.cc | 80 ComputeAndCompareR2<int32>(&builder, {{0, 0}, {0, 0}}, {});
|
| matrix_test.cc | 72 ComputeAndCompareR2<T>(&builder, kv.second, {a_data.get()});
|
| svd_test.cc | 148 ComputeAndCompareR2<float>(&builder, simple_2d_4x4_, {a_data.get()},
202 ComputeAndCompareR2<float>(&builder, singular_values, {a_data.get()},
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| binop_scaling_test.cc | 44 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001));
60 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001));
76 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001));
92 ComputeAndCompareR2<float>(&builder, *aexpected, {}, ErrorSpec(0.0001));
109 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001));
|
| transpose_test.cc | 44 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 0), {}, error_spec_);
52 ComputeAndCompareR2<float>(&builder, Array2D<float>(42, 0), {}, error_spec_);
60 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 7), {}, error_spec_);
73 ComputeAndCompareR2<float>(&builder, expected, {}, error_spec_);
129 ComputeAndCompareR2<float>(&builder, expected, {}, error_spec_);
142 ComputeAndCompareR2<float>(&builder, *expected, {}, ErrorSpec(1e-4));
154 ComputeAndCompareR2<float>(&builder, *expected, {}, ErrorSpec(1e-4));
|
| multidimensional_slice_test.cc | 42 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.000001));
|
| triangular_solve_test.cc | 114 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 10),
135 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()},
156 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()},
177 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()},
198 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()},
220 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()},
242 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()},
261 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()},
283 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()},
305 ComputeAndCompareR2<float>(&builder, expected, {a_data.get(), b_data.get()}
[all...] |
| cholesky_test.cc | 61 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()},
87 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()},
110 ComputeAndCompareR2<float>(&builder, expected, {a_data.get()},
|
| vector_ops_reduce_test.cc | 141 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_);
153 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_);
164 ComputeAndCompareR2<float>(&builder_, expected_array, {}, errspec_);
|
| slice_test.cc | 85 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 0), {});
93 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 5), {});
101 ComputeAndCompareR2<float>(&builder, Array2D<float>(2, 0), {});
122 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.000001));
136 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.000001));
154 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.000001));
405 ComputeAndCompareR2<int32>(&builder, *expected, {arg.get()});
|
| dot_operation_test.cc | 149 this->template ComputeAndCompareR2<T>(&builder, Array2D<T>(0, 0), {},
161 this->template ComputeAndCompareR2<T>(&builder, Array2D<T>(0, 3), {},
173 this->template ComputeAndCompareR2<T>(&builder, Array2D<T>(3, 0), {},
184 this->template ComputeAndCompareR2<T>(
212 this->template ComputeAndCompareR2<T>(
241 ComputeAndCompareR2<T>(&builder, expected,
359 ComputeAndCompareR2<NativeT>(&builder, *expected, args, error_spec);
491 ComputeAndCompareR2<T>(&builder, expected,
522 ComputeAndCompareR2<complex64>(
539 this->template ComputeAndCompareR2<T>(&builder, expected, {}
[all...] |
| concat_test.cc | 147 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 0), {},
164 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001));
178 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001));
188 ComputeAndCompareR2<float>(&builder, *b_array, {}, ErrorSpec(0.0001));
203 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001));
213 ComputeAndCompareR2<float>(&builder, *a_array, {}, ErrorSpec(0.0001));
234 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001));
358 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001));
379 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001));
405 ComputeAndCompareR2<float>(&builder, expected, {}, ErrorSpec(0.0001))
[all...] |
| select_and_scatter_test.cc | 257 ComputeAndCompareR2<int32>(&builder_, expected, {});
272 ComputeAndCompareR2<float>(&builder_, expected, {}, ErrorSpec(1e-7));
286 ComputeAndCompareR2<int32>(&builder_, expected, {});
298 ComputeAndCompareR2<int32>(&builder_, expected, {});
310 ComputeAndCompareR2<int32>(&builder_, expected, {});
324 ComputeAndCompareR2<int32>(&builder_, expected, {});
337 ComputeAndCompareR2<float>(&builder_, expected, {}, ErrorSpec(1e-7));
|
| broadcast_simple_test.cc | 118 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001));
130 ComputeAndCompareR2<float>(&b, expected, {param_data.get()},
138 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001));
145 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001));
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));
279 ComputeAndCompareR2<float>(&b, expected, {}, ErrorSpec(0.0001))
[all...] |
| constants_test.cc | 101 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 2), {}, error_spec_);
111 ComputeAndCompareR2<float>(&builder, *constant, {}, error_spec_);
|
| matrix_ops_simple_test.cc | 154 ComputeAndCompareR2<T>(&builder, expected, {}, error_spec);
235 ComputeAndCompareR2<T>(&builder, expected,
|
| pad_test.cc | 337 ComputeAndCompareR2<float>(&b, *expected, {}, DefaultErrorSpec());
359 ComputeAndCompareR2<float>(&b, *expected, {}, DefaultErrorSpec());
386 ComputeAndCompareR2<float>(&b, *expected, {}, DefaultErrorSpec());
414 ComputeAndCompareR2<float>(&b, *expected, {}, DefaultErrorSpec());
442 ComputeAndCompareR2<float>(&b, *expected, {}, DefaultErrorSpec());
|
| array_elementwise_ops_test.cc | 758 ComputeAndCompareR2<bool>(&builder, expected_array, {});
786 ComputeAndCompareR2<int32>(&builder, expected_array, {});
814 ComputeAndCompareR2<uint32>(&builder, expected_array, {});
842 ComputeAndCompareR2<bool>(&builder, expected_array, {});
870 ComputeAndCompareR2<int32>(&builder, expected_array, {});
898 ComputeAndCompareR2<uint32>(&builder, expected_array, {});
926 ComputeAndCompareR2<bool>(&builder, expected_array, {});
954 ComputeAndCompareR2<int32>(&builder, expected_array, {});
[all...] |
| tuple_test.cc | 130 ComputeAndCompareR2<float>(&builder, Array2D<float>(constant_matrix), {},
142 ComputeAndCompareR2<float>(&builder, Array2D<float>(0, 101), {}, error_spec_);
182 ComputeAndCompareR2<float>(&builder, expected, {}, error_spec_);
275 ComputeAndCompareR2<float>(&builder, expected, {}, error_spec_);
|
| params_test.cc | 102 ComputeAndCompareR2<float>(&builder, Array2D<float>(3, 0),
117 ComputeAndCompareR2<float>(&builder, expected_array, {param0_data.get()},
509 ComputeAndCompareR2(&builder, expected, {data.get()}, ErrorSpec(1e-3));
|
| client_library_test_base.h | 151 void ComputeAndCompareR2(XlaBuilder* builder,
155 void ComputeAndCompareR2(XlaBuilder* builder,
468 void ClientLibraryTestBase::ComputeAndCompareR2(
478 void ClientLibraryTestBase::ComputeAndCompareR2(
|
| unary_op_test.cc | 200 ComputeAndCompareR2<float>(&builder, {{0, 0}, {0, 0}}, {});
|
