Home
last modified time | relevance | path

Searched refs:dot_dnums (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Ddot_operation_test.cc927 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
928 dot_dnums.add_lhs_contracting_dimensions(1); in XLA_TEST_F()
929 dot_dnums.add_rhs_contracting_dimensions(0); in XLA_TEST_F()
930 DotGeneral(dynamic_slice, rhs_constant, dot_dnums); in XLA_TEST_F()
955 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
956 dot_dnums.add_lhs_contracting_dimensions(1); in XLA_TEST_F()
957 dot_dnums.add_rhs_contracting_dimensions(0); in XLA_TEST_F()
958 DotGeneral(lhs_constant, dynamic_slice, dot_dnums); in XLA_TEST_F()
985 DotDimensionNumbers dot_dnums; in XLA_TEST_F() local
986 dot_dnums.add_lhs_contracting_dimensions(0); in XLA_TEST_F()
[all …]
Dmultioutput_fusion_test.cc89 DotDimensionNumbers dot_dnums; in RunTest2D() local
90 dot_dnums.add_lhs_contracting_dimensions(1); in RunTest2D()
91 dot_dnums.add_rhs_contracting_dimensions(0); in RunTest2D()
93 elem_shape2, sub, add2, dot_dnums, DefaultPrecisionConfig(2))); in RunTest2D()
151 DotDimensionNumbers dot_dnums; in RunTest1D() local
152 dot_dnums.add_lhs_contracting_dimensions(0); in RunTest1D()
153 dot_dnums.add_rhs_contracting_dimensions(0); in RunTest1D()
156 dot_dnums, DefaultPrecisionConfig(2))); in RunTest1D()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc965 DotDimensionNumbers dot_dnums; in TEST_F() local
966 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
967 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
969 ShapeInference::InferDotOpShape(f32_, vector_32_, dot_dnums); in TEST_F()
977 DotDimensionNumbers dot_dnums; in TEST_F() local
978 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
979 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
981 ShapeUtil::MakeShape(F32, {32, 32, 32}), matrix_32_64_, dot_dnums); in TEST_F()
989 DotDimensionNumbers dot_dnums; in TEST_F() local
990 dot_dnums.add_lhs_contracting_dimensions(0); in TEST_F()
[all …]
Ddot_decomposer.cc136 DotDimensionNumbers dot_dnums; in DecomposeBatchDot() local
137 dot_dnums.add_lhs_contracting_dimensions(1); in DecomposeBatchDot()
138 dot_dnums.add_rhs_contracting_dimensions(0); in DecomposeBatchDot()
141 dot_dnums, dot->precision_config())); in DecomposeBatchDot()
269 DotDimensionNumbers dot_dnums; in CanonicalizeDot() local
271 dot_dnums.add_lhs_batch_dimensions(i); in CanonicalizeDot()
272 dot_dnums.add_rhs_batch_dimensions(i); in CanonicalizeDot()
274 dot_dnums.add_lhs_contracting_dimensions(num_batch_dims + 1); in CanonicalizeDot()
275 dot_dnums.add_rhs_contracting_dimensions(num_batch_dims); in CanonicalizeDot()
279 reshaped_lhs, reshaped_rhs, dot_dnums, original_dot->precision_config())); in CanonicalizeDot()
Dhlo_instruction_test.cc1149 DotDimensionNumbers dot_dnums; in TEST_F() local
1150 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
1151 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
1153 sout, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1190 DotDimensionNumbers dot_dnums; in TEST_F() local
1191 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
1192 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
1194 s, x, reshape, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
1241 DotDimensionNumbers dot_dnums; in TEST_F() local
1242 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
[all …]
Dhlo_computation_test.cc550 DotDimensionNumbers dot_dnums; in TEST_F() local
551 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
552 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
557 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config)); in TEST_F()
585 DotDimensionNumbers dot_dnums; in TEST_F() local
586 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
587 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
592 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config)); in TEST_F()
621 DotDimensionNumbers dot_dnums; in TEST_F() local
622 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
[all …]
Dheap_simulator_test.cc491 DotDimensionNumbers dot_dnums; in TEST_F() local
492 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
493 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
495 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
527 DotDimensionNumbers dot_dnums; in TEST_F() local
528 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
529 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
531 f32vec4_, mul, paramY, dot_dnums, DefaultPrecisionConfig(2))); in TEST_F()
565 DotDimensionNumbers dot_dnums; in TEST_F() local
566 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
[all …]
Dhlo_evaluator_test.cc735 DotDimensionNumbers dot_dnums; in TEST_P() local
736 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_P()
737 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_P()
739 rhs_instruction, dot_dnums, in TEST_P()
781 DotDimensionNumbers dot_dnums; in TEST_P() local
782 dot_dnums.add_lhs_contracting_dimensions(0); in TEST_P()
783 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_P()
785 rhs_instruction, dot_dnums, in TEST_P()
825 DotDimensionNumbers dot_dnums; in TEST_P() local
826 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_P()
[all …]
Dbfloat16_normalization_test.cc350 DotDimensionNumbers dot_dnums; in TEST_F() local
351 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
352 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
357 HloInstruction::CreateDot(bf16_shape, a, b, dot_dnums, precision_config)); in TEST_F()
Dalgebraic_simplifier_test.cc3767 DotDimensionNumbers dot_dnums; in TEST_F() local
3768 dot_dnums.add_lhs_batch_dimensions(0); in TEST_F()
3769 dot_dnums.add_rhs_batch_dimensions(0); in TEST_F()
3770 builder.AddInstruction(HloInstruction::CreateDot(r1f32, x, y, dot_dnums, in TEST_F()
4344 DotDimensionNumbers dot_dnums; in TEST_P() local
4345 dot_dnums.add_lhs_batch_dimensions(0); in TEST_P()
4346 dot_dnums.add_lhs_batch_dimensions(1); in TEST_P()
4347 dot_dnums.add_lhs_batch_dimensions(2); in TEST_P()
4348 dot_dnums.add_rhs_batch_dimensions(0); in TEST_P()
4349 dot_dnums.add_rhs_batch_dimensions(1); in TEST_P()
[all …]
Ddynamic_dimension_inference_test.cc258 DotDimensionNumbers dot_dnums; in TEST_F() local
259 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
260 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
262 HloInstruction::CreateDot(xz_shape, a_param, b_param, dot_dnums, in TEST_F()
Dtuple_points_to_analysis_test.cc1126 DotDimensionNumbers dot_dnums; in TEST_F() local
1127 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
1128 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
1133 HloInstruction::CreateDot(data_shape, a, b, dot_dnums, precision_config)); in TEST_F()
Dbfloat16_propagation_test.cc88 DotDimensionNumbers dot_dnums; in CreateDot() local
89 dot_dnums.add_lhs_contracting_dimensions(1); in CreateDot()
90 dot_dnums.add_rhs_contracting_dimensions(0); in CreateDot()
91 return HloInstruction::CreateDot(shape, lhs, rhs, dot_dnums, in CreateDot()
Dbuffer_assignment_test.cc1592 DotDimensionNumbers dot_dnums; in TEST_F() local
1593 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
1594 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
1599 shape_2x4, param_a, param_b, dot_dnums, precision_config)); in TEST_F()
1601 shape_3x4, param_b, param_c, dot_dnums, precision_config)); in TEST_F()
Dhlo_dataflow_analysis_test.cc2505 DotDimensionNumbers dot_dnums; in TEST_F() local
2506 dot_dnums.add_lhs_contracting_dimensions(1); in TEST_F()
2507 dot_dnums.add_rhs_contracting_dimensions(0); in TEST_F()
2512 HloInstruction::CreateDot(data_shape, a, b, dot_dnums, precision_config)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc39 DotDimensionNumbers dot_dnums; in MakeDot() local
40 dot_dnums.add_lhs_contracting_dimensions(1); in MakeDot()
41 dot_dnums.add_rhs_contracting_dimensions(0); in MakeDot()
45 return HloInstruction::CreateDot(shape, lhs, rhs, dot_dnums, in MakeDot()