Searched defs:dot_shape (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_layout_assignment_test.cc | 271 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); in RunDotOutputFusion() local
|
| H A D | cpu_instruction_fusion_test.cc | 622 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); in CreateComputationForDotAddOutputFusionTest() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | algebraic_simplifier_test.cc | 5833 Shape dot_shape = ShapeUtil::MakeShape(element_type, output_dims); in TEST_P() local
5889 Shape dot_shape = ShapeUtil::MakeShape(element_type, {m, n}); in TEST_P() local
5993 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); in TEST_P() local
6058 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); in TEST_P() local
6333 Shape dot_shape = ShapeUtil::MakeShape(F32, {dot_row_size, dot_col_size}); in TEST_P() local
6406 Shape dot_shape = ShapeUtil::MakeShape(F32, {dot_row_size, dot_col_size}); in TEST_P() local
|