| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 118 tf_default_dnums_for_backward_filter_, DefaultPrecisionConfig(2))); in TEST_F()
159 tf_default_dnums_for_backward_filter_, DefaultPrecisionConfig(2))); in TEST_F()
189 tf_default_dnums_for_backward_filter_, DefaultPrecisionConfig(2))); in TEST_F()
219 tf_default_dnums_for_backward_filter_, DefaultPrecisionConfig(2))); in TEST_F()
248 tf_default_dnums_for_backward_filter_, DefaultPrecisionConfig(2))); in TEST_F()
294 DefaultPrecisionConfig(2))); in TEST_F()
352 tf_default_dnums_for_backward_input_, DefaultPrecisionConfig(2))); in TEST_F()
387 tf_default_dnums_for_backward_input_, DefaultPrecisionConfig(2))); in TEST_F()
434 tf_default_dnums_for_backward_input_, DefaultPrecisionConfig(2))); in TEST_F()
485 tf_default_dnums_for_backward_input_, DefaultPrecisionConfig(2))); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | conv_canonicalization_test.cc | 88 conv_window_, dnums, DefaultPrecisionConfig(2))); in TEST_F()
151 conv_window_, dnums, DefaultPrecisionConfig(2))); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding_test.cc | 251 DefaultPrecisionConfig(2))); in TEST_F()
308 DefaultPrecisionConfig(2))); in TEST_F()
370 DefaultPrecisionConfig(2))); in TEST_F()
438 DefaultPrecisionConfig(2))); in TEST_F()
|
| D | hlo_instruction_test.cc | 1214 sout, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1313 s, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1364 data_shape, a, b_t, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1445 sout, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1624 sout, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1666 sout, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1722 sout, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
2032 shape, p0, p0, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
|
| D | heap_simulator_test.cc | 685 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
721 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
762 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
764 f32vec4_, dot0, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
803 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
805 f32vec4_, dot0, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
|
| D | hlo_evaluator_test.cc | 817 DefaultPrecisionConfig(2))); in TEST_P()
863 DefaultPrecisionConfig(2))); in TEST_P()
907 DefaultPrecisionConfig(2))); in TEST_P()
949 DefaultPrecisionConfig(2))); in TEST_P()
1006 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
1061 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
1145 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
1223 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
1283 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
1347 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
[all …]
|
| D | dynamic_dimension_inference_test.cc | 327 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
372 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
409 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
465 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
|
| D | dynamic_padder_test.cc | 287 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
330 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 1737 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_F()
3891 window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
4164 DefaultPrecisionConfig(2))); in TEST_F()
4539 DefaultPrecisionConfig(2))); in TEST_F()
5213 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
5271 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
5354 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
5419 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
5638 dot_shape, ds, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
5711 dot_shape, lhs, ds, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
|
| D | memory_space_assignment_test.cc | 4842 result_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4888 result_shape, lhs, bitcast, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4934 result_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4995 result_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
5035 result_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
5089 result_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
|
| D | bfloat16_propagation_test.cc | 92 DefaultPrecisionConfig(2)); in CreateDot()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multioutput_fusion_test.cc | 93 elem_shape2, sub, add2, dot_dnums, DefaultPrecisionConfig(2))); in RunTest2D()
156 dot_dnums, DefaultPrecisionConfig(2))); in RunTest1D()
|
| D | hlo_test_base.h | 102 static PrecisionConfig DefaultPrecisionConfig(int operands);
|
| D | hlo_test_base.cc | 161 PrecisionConfig HloTestBase::DefaultPrecisionConfig(int operands) { in DefaultPrecisionConfig() function in xla::HloTestBase
|