Searched refs:output_dimension_product (Results 1 – 1 of 1) sorted by relevance
1294 int64 input_dimension_product = 1, output_dimension_product = 1; in AlignLayouts() local1297 if (input_dimension_product == output_dimension_product && in AlignLayouts()1300 input_dimension_product = output_dimension_product = 1; in AlignLayouts()1302 if (input_dimension_product < output_dimension_product || in AlignLayouts()1311 output_dimension_product *= output_shape.dimensions(j); in AlignLayouts()1315 if (input_dimension_product != output_dimension_product) { in AlignLayouts()