| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_rewriter_test.cc | 117 tf_default_dnums_for_backward_filter_, DefaultPrecisionConfig(2))); in TEST_F()
157 tf_default_dnums_for_backward_filter_, DefaultPrecisionConfig(2))); in TEST_F()
187 tf_default_dnums_for_backward_filter_, DefaultPrecisionConfig(2))); in TEST_F()
217 tf_default_dnums_for_backward_filter_, DefaultPrecisionConfig(2))); in TEST_F()
246 tf_default_dnums_for_backward_filter_, DefaultPrecisionConfig(2))); in TEST_F()
292 DefaultPrecisionConfig(2))); in TEST_F()
347 tf_default_dnums_for_backward_input_, DefaultPrecisionConfig(2))); in TEST_F()
381 tf_default_dnums_for_backward_input_, DefaultPrecisionConfig(2))); in TEST_F()
428 tf_default_dnums_for_backward_input_, DefaultPrecisionConfig(2))); in TEST_F()
478 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 | 249 DefaultPrecisionConfig(2))); in TEST_F()
305 DefaultPrecisionConfig(2))); in TEST_F()
366 DefaultPrecisionConfig(2))); in TEST_F()
433 DefaultPrecisionConfig(2))); in TEST_F()
|
| D | dynamic_padder_test.cc | 131 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
174 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
|
| D | hlo_evaluator_test.cc | 740 DefaultPrecisionConfig(2))); in TEST_P()
786 DefaultPrecisionConfig(2))); in TEST_P()
830 DefaultPrecisionConfig(2))); in TEST_P()
872 DefaultPrecisionConfig(2))); in TEST_P()
929 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
984 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
1068 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
1146 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
1206 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
1270 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
[all …]
|
| D | heap_simulator_test.cc | 495 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
531 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
569 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
571 f32vec4_, dot0, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
610 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
612 f32vec4_, dot0, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
|
| D | hlo_instruction_test.cc | 1153 sout, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1194 s, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1245 data_shape, a, b_t, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1326 sout, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1491 sout, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1533 sout, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1589 sout, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
|
| D | dynamic_dimension_inference_test.cc | 263 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
315 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 1373 /*batch_group_count=*/1, window, dnums, DefaultPrecisionConfig(2))); in TEST_F()
3129 window, dnums, DefaultPrecisionConfig(2))); in TEST_P()
3395 DefaultPrecisionConfig(2))); in TEST_F()
3771 DefaultPrecisionConfig(2))); in TEST_F()
4356 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4414 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4490 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4555 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4678 dot_shape, ds, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4751 dot_shape, lhs, ds, 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 | 126 static PrecisionConfig DefaultPrecisionConfig(int operands);
|
| D | hlo_test_base.cc | 174 PrecisionConfig HloTestBase::DefaultPrecisionConfig(int operands) { in DefaultPrecisionConfig() function in xla::HloTestBase
|