| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| H A D | batch_dot.cc | 107 xla::DotDimensionNumbers dot_dnums; local
108 dot_dnums.add_lhs_contracting_dimensions(x_inner_dim);
109 dot_dnums.add_rhs_contracting_dimensions(y_inner_dim);
111 dot_dnums.add_lhs_batch_dimensions(batch_dimension_number);
112 dot_dnums.add_rhs_batch_dimensions(batch_dimension_number);
114 return builder->DotGeneral(x, y, dot_dnums);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | multioutput_fusion_test.cc | 81 DotDimensionNumbers dot_dnums; local
82 dot_dnums.add_lhs_contracting_dimensions(1);
83 dot_dnums.add_rhs_contracting_dimensions(0);
85 HloInstruction::CreateDot(elem_shape2, sub, add2, dot_dnums));
140 DotDimensionNumbers dot_dnums; local
141 dot_dnums.add_lhs_contracting_dimensions(0);
142 dot_dnums.add_rhs_contracting_dimensions(0);
144 ShapeUtil::MakeShape(F32, {1}), sub, reshape, dot_dnums));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shape_inference_test.cc | 902 DotDimensionNumbers dot_dnums; local
903 dot_dnums.add_lhs_contracting_dimensions(1);
904 dot_dnums.add_rhs_contracting_dimensions(0);
906 ShapeInference::InferDotOpShape(f32_, vector_32_, dot_dnums);
914 DotDimensionNumbers dot_dnums; local
915 dot_dnums.add_lhs_contracting_dimensions(1);
916 dot_dnums.add_rhs_contracting_dimensions(0);
918 ShapeUtil::MakeShape(F32, {32, 32, 32}), matrix_32_64_, dot_dnums);
926 DotDimensionNumbers dot_dnums; local
927 dot_dnums
940 DotDimensionNumbers dot_dnums; local
954 DotDimensionNumbers dot_dnums; local
968 DotDimensionNumbers dot_dnums; local
990 DotDimensionNumbers dot_dnums; local
1015 DotDimensionNumbers dot_dnums; local
1036 DotDimensionNumbers dot_dnums; local
1055 DotDimensionNumbers dot_dnums; local
1074 DotDimensionNumbers dot_dnums; local
1093 DotDimensionNumbers dot_dnums; local
[all...] |
| H A D | transpose_folding_test.cc | 67 DotDimensionNumbers dot_dnums; local
68 dot_dnums.add_lhs_contracting_dimensions(1);
69 dot_dnums.add_rhs_contracting_dimensions(0);
72 /*rhs=*/transpose_y, dot_dnums));
110 DotDimensionNumbers dot_dnums; local
111 dot_dnums.add_lhs_contracting_dimensions(1);
112 dot_dnums.add_rhs_contracting_dimensions(0);
115 /*lhs=*/transpose0, /*rhs=*/transpose1, dot_dnums));
178 DotDimensionNumbers dot_dnums; local
179 dot_dnums
[all...] |
| H A D | dot_decomposer.cc | 134 DotDimensionNumbers dot_dnums; local
135 dot_dnums.add_lhs_contracting_dimensions(1);
136 dot_dnums.add_rhs_contracting_dimensions(0);
138 dot_shape_r2, lhs_slice_r2, rhs_slice_r2, dot_dnums));
|
| H A D | graphviz_example.cc | 111 DotDimensionNumbers dot_dnums; local
112 dot_dnums.add_lhs_contracting_dimensions(1);
113 dot_dnums.add_rhs_contracting_dimensions(0);
115 HloInstruction::CreateDot(vshape, clamp, param_v0, dot_dnums));
|
| H A D | heap_simulator_test.cc | 262 DotDimensionNumbers dot_dnums; local
263 dot_dnums.add_lhs_contracting_dimensions(1);
264 dot_dnums.add_rhs_contracting_dimensions(0);
266 HloInstruction::CreateDot(f32vec4_, mul, paramY, dot_dnums));
298 DotDimensionNumbers dot_dnums; local
299 dot_dnums.add_lhs_contracting_dimensions(1);
300 dot_dnums.add_rhs_contracting_dimensions(0);
302 HloInstruction::CreateDot(f32vec4_, mul, paramY, dot_dnums));
336 DotDimensionNumbers dot_dnums; local
337 dot_dnums
377 DotDimensionNumbers dot_dnums; local
[all...] |
| H A D | liveness_util_test.cc | 279 DotDimensionNumbers dot_dnums; local
280 dot_dnums.add_lhs_contracting_dimensions(1);
281 dot_dnums.add_rhs_contracting_dimensions(0);
283 HloInstruction::CreateDot(data_shape, a, b, dot_dnums));
317 DotDimensionNumbers dot_dnums; local
318 dot_dnums.add_lhs_contracting_dimensions(1);
319 dot_dnums.add_rhs_contracting_dimensions(0);
321 HloInstruction::CreateDot(data_shape, a, b_t, dot_dnums));
|
| H A D | hlo_instruction_test.cc | 1096 DotDimensionNumbers dot_dnums; local
1097 dot_dnums.add_lhs_contracting_dimensions(1);
1098 dot_dnums.add_rhs_contracting_dimensions(0);
1100 HloInstruction::CreateDot(sout, x, reshape, dot_dnums));
1154 DotDimensionNumbers dot_dnums; local
1155 dot_dnums.add_lhs_contracting_dimensions(1);
1156 dot_dnums.add_rhs_contracting_dimensions(0);
1158 HloInstruction::CreateDot(data_shape, a, b_t, dot_dnums));
1235 DotDimensionNumbers dot_dnums; local
1236 dot_dnums
[all...] |
| H A D | hlo_evaluator_test.cc | 634 DotDimensionNumbers dot_dnums; local
635 dot_dnums.add_lhs_contracting_dimensions(1);
636 dot_dnums.add_rhs_contracting_dimensions(0);
638 rhs_instruction, dot_dnums));
679 DotDimensionNumbers dot_dnums; local
680 dot_dnums.add_lhs_contracting_dimensions(0);
681 dot_dnums.add_rhs_contracting_dimensions(0);
683 rhs_instruction, dot_dnums));
722 DotDimensionNumbers dot_dnums; local
723 dot_dnums
[all...] |
| H A D | algebraic_simplifier_test.cc | 2338 DotDimensionNumbers dot_dnums; local
2339 dot_dnums.add_lhs_contracting_dimensions(1);
2340 dot_dnums.add_rhs_contracting_dimensions(0);
2341 builder.AddInstruction(HloInstruction::CreateDot(r1f32, x, y, dot_dnums));
2601 DotDimensionNumbers dot_dnums; local
2602 dot_dnums.add_lhs_contracting_dimensions(1);
2603 dot_dnums.add_rhs_contracting_dimensions(0);
2605 HloInstruction::CreateDot(dot_shape, lhs, rhs, dot_dnums));
2675 DotDimensionNumbers dot_dnums; local
2676 dot_dnums
2739 DotDimensionNumbers dot_dnums; local
[all...] |
| H A D | buffer_assignment_test.cc | 1422 DotDimensionNumbers dot_dnums; local
1423 dot_dnums.add_lhs_contracting_dimensions(1);
1424 dot_dnums.add_rhs_contracting_dimensions(0);
1426 HloInstruction::CreateDot(shape_2x4, param_a, param_b, dot_dnums));
1428 HloInstruction::CreateDot(shape_3x4, param_b, param_c, dot_dnums));
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_instruction_fusion_test.cc | 36 DotDimensionNumbers dot_dnums; local
37 dot_dnums.add_lhs_contracting_dimensions(1);
38 dot_dnums.add_rhs_contracting_dimensions(0);
39 return HloInstruction::CreateDot(shape, lhs, rhs, dot_dnums);
|