Home
last modified time | relevance | path

Searched refs:ComputeAndCompareR1 (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvert_test.cc54 ComputeAndCompareR1<int32>(&builder, expected, {}); in TEST_F()
63 ComputeAndCompareR1<uint32>(&builder, expected, {}); in TEST_F()
72 ComputeAndCompareR1<bool>(&builder, expected, {}); in TEST_F()
81 ComputeAndCompareR1<uint32>(&builder, expected, {}); in TEST_F()
90 ComputeAndCompareR1<int32>(&builder, expected, {}); in TEST_F()
99 ComputeAndCompareR1<bool>(&builder, expected, {}); in TEST_F()
108 ComputeAndCompareR1<float>(&builder, expected, {}); in TEST_F()
117 ComputeAndCompareR1<bool>(&builder, expected, {}); in TEST_F()
126 ComputeAndCompareR1<float>(&builder, expected, {}); in TEST_F()
135 ComputeAndCompareR1<int32>(&builder, expected, {}); in TEST_F()
[all …]
Darray_elementwise_ops_test.cc57 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
65 ComputeAndCompareR1<float>(&builder, {2.5f, -3.14f, -2.25f, 10.0f, -6.0f}, {}, in XLA_TEST_F()
79 ComputeAndCompareR1<int32>(&builder, in XLA_TEST_F()
90 ComputeAndCompareR1<complex64>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
99 ComputeAndCompareR1<complex64>( in XLA_TEST_F()
119 ComputeAndCompareR1<int64>(&builder, in XLA_TEST_F()
137 ComputeAndCompareR1<bool>(&builder, {}, {}); in XLA_TEST_F()
171 ComputeAndCompareR1<bool>(&builder, {false, true, false, true, false, false}, in XLA_TEST_F()
181 ComputeAndCompareR1<float>(&builder, {97.5f, 6.27f, 5.0f, 0.5f, -993.0f}, {}, in XLA_TEST_F()
191 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
[all …]
Dvector_ops_simple_test.cc61 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
81 ComputeAndCompareR1<float>(&builder, expected, {}, in XLA_TEST_F()
116 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
125 ComputeAndCompareR1<int32>(&builder, expected, {}); in XLA_TEST_F()
135 ComputeAndCompareR1<uint32>(&builder, expected, {}); in XLA_TEST_F()
147 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
162 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
175 ComputeAndCompareR1<float>(&builder, expected, {}); in XLA_TEST_F()
191 ComputeAndCompareR1<float>(&builder, {41.0f, 22.0f, 23.0f, 84.0f}, in XLA_TEST_F()
231 ComputeAndCompareR1<float>(&builder, expected_vec, in XLA_TEST_F()
[all …]
Dselect_test.cc73 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
85 ComputeAndCompareR1<float>(&builder, {10.0f, 25.5f, 1.0f, -10.0f, -6.0f}, {}, in TEST_F()
100 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
116 ComputeAndCompareR1<float>(&builder, {10.0f, 25.5f, 1.0f, -10.0f, -6.0f}, {}, in TEST_F()
132 ComputeAndCompareR1<float>(&builder, {-2.5f, 25.5f, 1.0f, 10.0f, 6.0f}, {}, in TEST_F()
151 ComputeAndCompareR1<float>(&builder, {41.0f, 22.0f, 23.0f, 84.0f}, in TEST_F()
192 ComputeAndCompareR1<float>(&builder, expected_vec, in TEST_F()
210 ComputeAndCompareR1<float>(&builder, {11.0f, -222.0f, 33.0f, -444.0f}, {}, in TEST_F()
227 ComputeAndCompareR1<float>(&builder, {-111.0f, -222.0f, 33.0f, 44.0f}, {}, in TEST_F()
239 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in XLA_TEST_F()
[all …]
Dbitcast_convert_test.cc50 ComputeAndCompareR1<int32>(&builder, expected, {}); in TEST_F()
59 ComputeAndCompareR1<float>(&builder, expected, {}); in TEST_F()
71 ComputeAndCompareR1<float>(&builder, expected, {}); in TEST_F()
80 ComputeAndCompareR1<float>(&builder, expected, {}); in XLA_TEST_F()
89 ComputeAndCompareR1<int32>(&builder, expected, {}); in TEST_F()
99 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0, 0)); in TEST_F()
111 ComputeAndCompareR1<int32>(&builder, expected, {}); in TEST_F()
123 ComputeAndCompareR1<float>(&builder, expected, {}); in TEST_F()
Dconstants_test.cc48 ComputeAndCompareR1<float>(&builder, {}, {}, error_spec_); in TEST_F()
57 ComputeAndCompareR1<float>(&builder, constant, {}, error_spec_); in TEST_F()
66 ComputeAndCompareR1<int32>(&builder, constant, {}); in TEST_F()
75 ComputeAndCompareR1<uint32>(&builder, constant, {}); in TEST_F()
84 ComputeAndCompareR1<float>(&builder, constant, {}, error_spec_); in TEST_F()
94 ComputeAndCompareR1<float>(&builder, constant, {}, error_spec_); in TEST_F()
189 ComputeAndCompareR1<float>(&b, {10, 11, 12}, {}, error_spec_); in TEST_F()
Dunary_op_test.cc45 ComputeAndCompareR1<float>(&builder, {}, {}); in AbsSize0TestHelper()
47 ComputeAndCompareR1<T>(&builder, {}, {}); in AbsSize0TestHelper()
57 ComputeAndCompareR1<T>(&builder, {2, 25, 0, 123, inf<T>(), inf<T>()}, {}); in AbsTestHelper()
67 ComputeAndCompareR1<T>( in SignTestHelper()
80 ComputeAndCompareR1<T>(&builder, {0, 0, 0, 0}, {}); in SignAbsTestHelper()
209 ComputeAndCompareR1<int32>(&builder, {0, 1}, {}); in XLA_TEST_F()
218 ComputeAndCompareR1<float>(&builder, {0.0, 1.0}, {}); in XLA_TEST_F()
Dconcat_test.cc56 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
65 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
88 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
98 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
108 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
118 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
137 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
280 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
325 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
337 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
[all …]
Dvector_ops_reduce_test.cc116 ComputeAndCompareR1<float>(&builder_, {6.0, 15.0}, {}, errspec_); in TEST_F()
130 ComputeAndCompareR1<float>(&builder_, {5.0, 7.0, 9.0}, {}, errspec_); in TEST_F()
173 ComputeAndCompareR1<float>(&builder_, {21.0, 21.0, 21.0}, {}, errspec_); in TEST_F()
182 ComputeAndCompareR1<float>(&builder_, {18.0, 45.0}, {}, errspec_); in XLA_TEST_F()
191 ComputeAndCompareR1<float>(&builder_, {15.0, 21.0, 27.0}, {}, errspec_); in TEST_F()
Dreduce_precision_test.cc239 ComputeAndCompareR1<float>(&builder, expected_values, {a_data.get()}); in XLA_TEST_P()
276 ComputeAndCompareR1<float>(&builder, {0.0f}, {a_data.get()}); in XLA_TEST_F()
302 ComputeAndCompareR1<float>(&builder, {-1.00001f, -1.00001f}, {a_data.get()}); in XLA_TEST_F()
326 ComputeAndCompareR1<float>(&builder, {-1.0f, -1.0f}, {a_data.get()}); in XLA_TEST_F()
351 ComputeAndCompareR1<float>(&builder, {-1.00001f}, {a_data.get()}); in XLA_TEST_F()
376 ComputeAndCompareR1<float>(&builder, {-1.0f, -1.0f}, {a_data.get()}); in XLA_TEST_F()
Daxpy_simple_test.cc40 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in TEST_F()
52 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
72 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in TEST_F()
Dmap_test.cc193 ComputeAndCompareR1<float>(&builder, {}, {param0_data.get()}, in XLA_TEST_F()
208 ComputeAndCompareR1<float>(&builder, {3.2f, 4.3f, 5.4f, 6.5f}, in TEST_F()
222 ComputeAndCompareR1<int32>(&builder, {1, 1, 1, 1}, {param0_data.get()}); in TEST_F()
235 ComputeAndCompareR1<uint32>(&builder, {1, 1, 1, 1}, {param0_data.get()}); in TEST_F()
249 ComputeAndCompareR1<float>( in TEST_F()
266 ComputeAndCompareR1<float>(&builder, {}, {param0_data.get()}, in XLA_TEST_F()
283 ComputeAndCompareR1<float>(&builder, {6.4f, 8.6f, 10.8f, 13.0f}, in TEST_F()
359 ComputeAndCompareR1<float>(&builder, {7.3f, 7.7, 4.3f, 0}, in TEST_F()
434 ComputeAndCompareR1<float>( in TEST_F()
446 ComputeAndCompareR1<bool>(&b, {false, true}, {}); in TEST_F()
[all …]
Dfmax_fmin_test.cc40 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in TEST_F()
60 ComputeAndCompareR1<float>(&builder, expected, in TEST_F()
82 ComputeAndCompareR1<float>(&builder, expected, in TEST_F()
Dreduce_test.cc198 ComputeAndCompareR1<bool>(&builder, expected, {input_global_data.get()}); in RunR2ToR1PredTest()
253 ComputeAndCompareR1<float>(&builder, expected, {input_global_data.get()}, in RunR2ToR1Test()
263 ComputeAndCompareR1<NativeT>(builder, expected, arguments, in ComputeAndCompareGeneric()
273 ComputeAndCompareR1<NativeT>(builder, expected, arguments); in ComputeAndCompareGeneric()
463 ComputeAndCompareR1<float>(&builder, expected, {input_global_data.get()}, in XLA_TEST_F()
494 ComputeAndCompareR1<float>(&builder, expected, {input_global_data.get()}, in XLA_TEST_F()
542 ComputeAndCompareR1<float>(&builder, expected, {input_global_data.get()}, in XLA_TEST_F()
647 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
668 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
678 ComputeAndCompareR1<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
[all …]
Dselect_and_scatter_test.cc211 ComputeAndCompareR1<float>(&builder_, {}, {}, ErrorSpec(1e-7)); in XLA_TEST_F()
223 ComputeAndCompareR1<float>(&builder_, expected, {}, ErrorSpec(1e-7)); in XLA_TEST_F()
234 ComputeAndCompareR1<int32>(&builder_, expected, {}); in XLA_TEST_F()
245 ComputeAndCompareR1<int32>(&builder_, expected, {}); in XLA_TEST_F()
445 ComputeAndCompareR1<float>(&builder_, expected, {}, ErrorSpec(1e-7)); in XLA_TEST_F()
457 ComputeAndCompareR1<float>(&builder_, expected, {}, ErrorSpec(1e-7)); in XLA_TEST_F()
Diota_test.cc44 ComputeAndCompareR1<float>(&builder, GetR1Expected<float>(num_elements), {}, in TEST_P()
47 ComputeAndCompareR1<uint32>(&builder, GetR1Expected<uint32>(num_elements), in TEST_P()
51 ComputeAndCompareR1<int32>(&builder, GetR1Expected<int32>(num_elements), in TEST_P()
Dparams_test.cc63 ComputeAndCompareR1<float>(&builder, {}, {param0_data.get()}, in XLA_TEST_F()
75 ComputeAndCompareR1<float>(&builder, {3.14f, -100.25f}, {param0_data.get()}, in XLA_TEST_F()
147 ComputeAndCompareR1<float>(&builder, {110, 440}, in XLA_TEST_F()
179 ComputeAndCompareR1<float>(&builder, {10, 20}, in XLA_TEST_F()
206 ComputeAndCompareR1<float>( in XLA_TEST_F()
246 ComputeAndCompareR1<float>(&builder, sum, param_data, ErrorSpec(0.0001f)); in XLA_TEST_F()
453 ComputeAndCompareR1<float>(&builder, expected, arguments, ErrorSpec(1e-5)); in XLA_TEST_F()
Dhalf_test.cc67 ComputeAndCompareR1<half>(&builder, expected, {x_data.get()}, error_spec_); in XLA_TEST_P()
122 ComputeAndCompareR1<bool>(&builder, expected, {x_data.get()}); in XLA_TEST_P()
161 ComputeAndCompareR1<half>(&builder, expected, {x_data.get(), y_data.get()}, in XLA_TEST_P()
216 ComputeAndCompareR1<bool>(&builder, expected, {x_data.get(), y_data.get()}); in XLA_TEST_P()
Dcall_test.cc93 ComputeAndCompareR1<float>(&builder, {}, {}, ErrorSpec(0.01f)); in XLA_TEST_F()
105 ComputeAndCompareR1<float>(&builder, {3.0f, 5.0f}, {}, ErrorSpec(0.01f)); in XLA_TEST_F()
Dreshape_motion_test.cc53 ComputeAndCompareR1<int32>(&builder, {34, 57, 115, 203, 341, 481}, {}); in TEST_F()
Dclient_library_test_base.h135 void ComputeAndCompareR1(XlaBuilder* builder,
139 void ComputeAndCompareR1(XlaBuilder* builder,
146 void ComputeAndCompareR1(XlaBuilder* builder,
443 void ClientLibraryTestBase::ComputeAndCompareR1( in ComputeAndCompareR1() function
452 void ClientLibraryTestBase::ComputeAndCompareR1( in ComputeAndCompareR1() function
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dcomparators_test.cc96 ComputeAndCompareR1<bool>(builder(), expected, {}); in XLA_TEST_F()
103 ComputeAndCompareR1<bool>(builder(), expected, {}); in XLA_TEST_F()
110 ComputeAndCompareR1<bool>(builder(), expected, {}); in XLA_TEST_F()
117 ComputeAndCompareR1<bool>(builder(), expected, {}); in XLA_TEST_F()
124 ComputeAndCompareR1<bool>(builder(), expected, {}); in XLA_TEST_F()
131 ComputeAndCompareR1<bool>(builder(), expected, {}); in XLA_TEST_F()
138 ComputeAndCompareR1<bool>(builder(), expected, {}); in XLA_TEST_F()
145 ComputeAndCompareR1<bool>(builder(), expected, {}); in XLA_TEST_F()
Dsorting_test.cc36 ComputeAndCompareR1<float>(&builder, {7.0, 6.0, 5.0}, {}); in XLA_TEST_F()
44 ComputeAndCompareR1<int>(&builder, {0, 1, 2}, {}); in XLA_TEST_F()
54 ComputeAndCompareR1<int>(&builder, {2, 1, 0}, {}); in XLA_TEST_F()
66 ComputeAndCompareR1<int>(&builder, {0, 2, 1}, {}); in XLA_TEST_F()
81 ComputeAndCompareR1<float>(&builder, inputs, {}); in XLA_TEST_F()
89 ComputeAndCompareR1<int>(&builder, {2, 3, 0, 1, 4}, {a_data.get()}); in XLA_TEST_F()
Dmath_test.cc45 ComputeAndCompareR1<T>(&b, in TestLogEdgeCases()
56 ComputeAndCompareR1<T>( in TestLog1pEdgeCases()
136 ComputeAndCompareR1<T>(&b, expected, {}, error_spec_); in TestSqrtPowInequivalence()
149 ComputeAndCompareR1<T>(&b, expected, {}, error_spec_); in TestErfEdgeCases()
235 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
247 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
255 ComputeAndCompareR1<float>(&builder, {0, 0}, {}, error_spec_); in XLA_TEST_F()
264 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
287 ComputeAndCompareR1<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
309 ComputeAndCompareR1<half>(&b, expected, {}, ErrorSpec{0.1}); in XLA_TEST_F()
[all …]
Darithmetic_test.cc36 ComputeAndCompareR1<int32>(&builder, expected, {}); in XLA_TEST_F()
45 ComputeAndCompareR1<int32>(&builder, expected, {}); in XLA_TEST_F()
54 ComputeAndCompareR1<int32>(&builder, expected, {}); in XLA_TEST_F()
63 ComputeAndCompareR1<int32>(&builder, expected, {}); in XLA_TEST_F()

12