| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_variants_test.cc | 66 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
81 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
98 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
113 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
128 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
143 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
158 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
173 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
188 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
209 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
[all …]
|
| D | pad_test.cc | 137 ComputeAndCompareR4<float>(&b, Array4D<float>(5, 2, 3, 2, 1.5f), {}, in XLA_TEST_P()
162 ComputeAndCompareR4<float>(&b, *expected, {}, DefaultErrorSpec()); in TEST_P()
182 ComputeAndCompareR4<float>(&b, *expected, {}, ErrorSpec(0.0001)); in TEST_P()
224 ComputeAndCompareR4<float>(&b, expected_array, {}, ErrorSpec(0.0001)); in TEST_P()
267 ComputeAndCompareR4<float>(&b, expected_array, {}, ErrorSpec(0.0001)); in XLA_TEST_P()
291 ComputeAndCompareR4<uint8>(&b, *expected, {}); in XLA_TEST_F()
318 ComputeAndCompareR4<int32>(&b, *expected, {}); in XLA_TEST_F()
|
| D | reverse_test.cc | 141 ComputeAndCompareR4<uint8>(&b, expected, {}); in XLA_TEST_F()
183 ComputeAndCompareR4<float>(&b, expected, {}, ErrorSpec(0.0001)); in TEST_F()
|
| D | select_and_scatter_test.cc | 362 ComputeAndCompareR4<float>(&builder_, e, {}, ErrorSpec(1e-7)); in TEST_F()
387 ComputeAndCompareR4<float>(&builder_, e, {}, ErrorSpec(1e-7)); in TEST_F()
412 ComputeAndCompareR4<float>(&builder_, e, {}, ErrorSpec(1e-7)); in TEST_F()
435 ComputeAndCompareR4<float>(&builder_, *e, {}, ErrorSpec(1e-7)); in TEST_F()
|
| D | constants_test.cc | 151 ComputeAndCompareR4<float>(&builder, input_array, {}, error_spec_); in TEST_F()
157 ComputeAndCompareR4<float>(&builder, input_array, {}, error_spec_); in TEST_F()
|
| D | client_library_test_base.h | 171 void ComputeAndCompareR4(XlaBuilder* builder,
175 void ComputeAndCompareR4(XlaBuilder* builder,
522 void ClientLibraryTestBase::ComputeAndCompareR4( in ComputeAndCompareR4() function
532 void ClientLibraryTestBase::ComputeAndCompareR4( in ComputeAndCompareR4() function
|
| D | binop_scaling_test.cc | 136 ComputeAndCompareR4<int>(&builder, expected, {}); in TEST_F()
|
| D | batch_normalization_test.cc | 116 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
133 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
228 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
651 ComputeAndCompareR4<float>( in XLA_TEST_P()
|
| D | convolution_dimension_numbers_test.cc | 127 ComputeAndCompareR4<float>(&builder, *expected_conv2, {weight_data.get()}, in XLA_TEST_F()
|
| D | slice_test.cc | 166 ComputeAndCompareR4(&builder, *expected, {}, ErrorSpec(0.000001)); in TEST_F()
494 ComputeAndCompareR4(&builder, *expected, {arg.get()}, ErrorSpec(0.000001)); in Run()
|
| D | array_elementwise_ops_test.cc | 1777 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
1788 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
2012 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
2024 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
2854 ComputeAndCompareR4<float>(&builder, *expected_4d, {}, error_spec_);
2882 ComputeAndCompareR4<float>(&builder, *expected_4d, {}, error_spec_); in XLA_TEST_F()
2911 ComputeAndCompareR4<float>(&builder, r4, {}, error_spec_); in XLA_TEST_F()
|
| D | vector_ops_simple_test.cc | 104 ComputeAndCompareR4<float>(&builder, expected, {}, in XLA_TEST_F()
|
| D | dot_operation_test.cc | 595 this->template ComputeAndCompareR4<T>( in XLA_TYPED_TEST()
745 this->template ComputeAndCompareR4<T>( in XLA_TYPED_TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling_test.cc | 75 ComputeAndCompareR4<float>(&builder, {{{{5, 4}}}}, {}, error_spec_); in XLA_TEST_F()
88 ComputeAndCompareR4<float>(&builder, {{{{5, 4, 5}}}}, {}, error_spec_); in XLA_TEST_F()
101 ComputeAndCompareR4<float>(&builder, {{{{5, 4, 4, 5, 5}, {5, 4, 3, 2, 1}}}}, in XLA_TEST_F()
118 ComputeAndCompareR4<float>(&builder, {{{{3, 3}}}}, {}, error_spec_); in XLA_TEST_F()
134 ComputeAndCompareR4<float>(&builder, {{{{3, 3, 3}}}}, {}, error_spec_); in XLA_TEST_F()
150 ComputeAndCompareR4<float>(&builder, in XLA_TEST_F()
166 ComputeAndCompareR4<float>(&builder, {{{{3, 3}}}}, {}, error_spec_); in XLA_TEST_F()
181 ComputeAndCompareR4<float>(&builder, {{{{1.5, 3, 4.5}, {3, 3, 3}}}}, {}, in XLA_TEST_F()
196 ComputeAndCompareR4<float>( in XLA_TEST_F()
214 ComputeAndCompareR4<float>( in XLA_TEST_F()
[all …]
|