| /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 …]
|
| /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 | 142 ComputeAndCompareR4<uint8>(&b, expected, {}); in XLA_TEST_F()
184 ComputeAndCompareR4<float>(&b, expected, {}, ErrorSpec(0.0001)); in TEST_F()
|
| D | select_and_scatter_test.cc | 373 ComputeAndCompareR4<float>(&builder_, e, {}, ErrorSpec(1e-7)); in TEST_F()
398 ComputeAndCompareR4<float>(&builder_, e, {}, ErrorSpec(1e-7)); in TEST_F()
423 ComputeAndCompareR4<float>(&builder_, e, {}, ErrorSpec(1e-7)); in TEST_F()
446 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 | binop_scaling_test.cc | 136 ComputeAndCompareR4<int>(&builder, expected, {}); in TEST_F()
|
| D | convolution_dimension_numbers_test.cc | 127 ComputeAndCompareR4<float>(&builder, *expected_conv2, {weight_data.get()}, in XLA_TEST_F()
|
| D | client_library_test_base.h | 173 void ComputeAndCompareR4(XlaBuilder* builder,
177 void ComputeAndCompareR4(XlaBuilder* builder,
553 void ClientLibraryTestBase::ComputeAndCompareR4( in ComputeAndCompareR4() function
563 void ClientLibraryTestBase::ComputeAndCompareR4( in ComputeAndCompareR4() function
|
| D | batch_normalization_test.cc | 117 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
134 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
229 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_P()
749 ComputeAndCompareR4<float>( in XLA_TEST_P()
|
| D | slice_test.cc | 166 ComputeAndCompareR4(&builder, *expected, {}, ErrorSpec(0.000001)); in TEST_F()
505 ComputeAndCompareR4(&builder, *expected, {arg.get()}, ErrorSpec(0.000001)); in Run()
|
| D | array_elementwise_ops_test.cc | 1883 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
1894 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
2118 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
2130 ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_); in XLA_TEST_F()
2960 ComputeAndCompareR4<float>(&builder, *expected_4d, {}, error_spec_);
2988 ComputeAndCompareR4<float>(&builder, *expected_4d, {}, error_spec_); in XLA_TEST_F()
3017 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 | 639 this->template ComputeAndCompareR4<T>( in XLA_TYPED_TEST()
785 this->template ComputeAndCompareR4<T>( in XLA_TYPED_TEST()
|