| /external/tensorflow/tensorflow/compiler/xla/service/ |
| dot_decomposer.cc | 136 DotDimensionNumbers dot_dnums; local
137 dot_dnums.add_lhs_contracting_dimensions(1);
138 dot_dnums.add_rhs_contracting_dimensions(0);
141 dot_dnums, dot->precision_config()));
269 DotDimensionNumbers dot_dnums; local
271 dot_dnums.add_lhs_batch_dimensions(i);
272 dot_dnums.add_rhs_batch_dimensions(i);
274 dot_dnums.add_lhs_contracting_dimensions(num_batch_dims + 1);
275 dot_dnums.add_rhs_contracting_dimensions(num_batch_dims);
279 reshaped_lhs, reshaped_rhs, dot_dnums, original_dot->precision_config()))
[all...] |
| shape_inference_test.cc | 965 DotDimensionNumbers dot_dnums; local
977 DotDimensionNumbers dot_dnums; local
989 DotDimensionNumbers dot_dnums; local
1003 DotDimensionNumbers dot_dnums; local
1017 DotDimensionNumbers dot_dnums; local
1031 DotDimensionNumbers dot_dnums; local
1053 DotDimensionNumbers dot_dnums; local
1077 DotDimensionNumbers dot_dnums; local
1098 DotDimensionNumbers dot_dnums; local
1118 DotDimensionNumbers dot_dnums; local
1137 DotDimensionNumbers dot_dnums; local
1156 DotDimensionNumbers dot_dnums; local
1175 DotDimensionNumbers dot_dnums; local
[all...] |
| hlo_computation_test.cc | 550 DotDimensionNumbers dot_dnums; local
551 dot_dnums.add_lhs_contracting_dimensions(1);
552 dot_dnums.add_rhs_contracting_dimensions(0);
557 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config));
585 DotDimensionNumbers dot_dnums; local
586 dot_dnums.add_lhs_contracting_dimensions(1);
587 dot_dnums.add_rhs_contracting_dimensions(0);
592 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config));
621 DotDimensionNumbers dot_dnums; local
622 dot_dnums.add_lhs_contracting_dimensions(1)
[all...] |
| hlo_instruction_test.cc | 1149 DotDimensionNumbers dot_dnums; local
1190 DotDimensionNumbers dot_dnums; local
1241 DotDimensionNumbers dot_dnums; local
1322 DotDimensionNumbers dot_dnums; local
1487 DotDimensionNumbers dot_dnums; local
1529 DotDimensionNumbers dot_dnums; local
1585 DotDimensionNumbers dot_dnums; local
[all...] |
| heap_simulator_test.cc | 491 DotDimensionNumbers dot_dnums; local
492 dot_dnums.add_lhs_contracting_dimensions(1);
493 dot_dnums.add_rhs_contracting_dimensions(0);
495 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2)));
527 DotDimensionNumbers dot_dnums; local
528 dot_dnums.add_lhs_contracting_dimensions(1);
529 dot_dnums.add_rhs_contracting_dimensions(0);
531 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2)));
565 DotDimensionNumbers dot_dnums; local
566 dot_dnums.add_lhs_contracting_dimensions(1)
606 DotDimensionNumbers dot_dnums; local
[all...] |
| bfloat16_normalization_test.cc | 350 DotDimensionNumbers dot_dnums; local
351 dot_dnums.add_lhs_contracting_dimensions(1);
352 dot_dnums.add_rhs_contracting_dimensions(0);
357 HloInstruction::CreateDot(bf16_shape, a, b, dot_dnums, precision_config));
|
| hlo_evaluator_test.cc | 735 DotDimensionNumbers dot_dnums; local
736 dot_dnums.add_lhs_contracting_dimensions(1);
737 dot_dnums.add_rhs_contracting_dimensions(0);
739 rhs_instruction, dot_dnums,
781 DotDimensionNumbers dot_dnums; local
782 dot_dnums.add_lhs_contracting_dimensions(0);
783 dot_dnums.add_rhs_contracting_dimensions(0);
785 rhs_instruction, dot_dnums,
825 DotDimensionNumbers dot_dnums; local
862 DotDimensionNumbers dot_dnums; local
[all...] |
| algebraic_simplifier_test.cc | 3767 DotDimensionNumbers dot_dnums; local
4344 DotDimensionNumbers dot_dnums; local
4410 DotDimensionNumbers dot_dnums; local
4484 DotDimensionNumbers dot_dnums; local
4549 DotDimensionNumbers dot_dnums; local
4670 DotDimensionNumbers dot_dnums; local
4743 DotDimensionNumbers dot_dnums; local
[all...] |
| dynamic_dimension_inference_test.cc | 258 DotDimensionNumbers dot_dnums; local
259 dot_dnums.add_lhs_contracting_dimensions(1);
260 dot_dnums.add_rhs_contracting_dimensions(0);
262 HloInstruction::CreateDot(xz_shape, a_param, b_param, dot_dnums,
|
| bfloat16_propagation_test.cc | 88 DotDimensionNumbers dot_dnums; local
89 dot_dnums.add_lhs_contracting_dimensions(1);
90 dot_dnums.add_rhs_contracting_dimensions(0);
91 return HloInstruction::CreateDot(shape, lhs, rhs, dot_dnums,
[all...] |
| tuple_points_to_analysis_test.cc | 1126 DotDimensionNumbers dot_dnums; local
[all...] |
| buffer_assignment_test.cc | [all...] |
| hlo_dataflow_analysis_test.cc | 2505 DotDimensionNumbers dot_dnums; local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| dot_operation_test.cc | 927 DotDimensionNumbers dot_dnums; local
955 DotDimensionNumbers dot_dnums; local
985 DotDimensionNumbers dot_dnums; local
1013 DotDimensionNumbers dot_dnums; local
1046 DotDimensionNumbers dot_dnums; local
1079 DotDimensionNumbers dot_dnums; local
1104 DotDimensionNumbers dot_dnums; local
1129 DotDimensionNumbers dot_dnums; local
1148 DotDimensionNumbers dot_dnums; local
[all...] |
| multioutput_fusion_test.cc | 89 DotDimensionNumbers dot_dnums; local
90 dot_dnums.add_lhs_contracting_dimensions(1);
91 dot_dnums.add_rhs_contracting_dimensions(0);
93 elem_shape2, sub, add2, dot_dnums, DefaultPrecisionConfig(2)));
151 DotDimensionNumbers dot_dnums; local
152 dot_dnums.add_lhs_contracting_dimensions(0);
153 dot_dnums.add_rhs_contracting_dimensions(0);
156 dot_dnums, DefaultPrecisionConfig(2)));
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion_test.cc | 39 DotDimensionNumbers dot_dnums; local
40 dot_dnums.add_lhs_contracting_dimensions(1);
41 dot_dnums.add_rhs_contracting_dimensions(0);
45 return HloInstruction::CreateDot(shape, lhs, rhs, dot_dnums,
[all...] |
