Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc1483 int64 input_dimension_product = 1, output_dimension_product = 1; in AlignLayouts() local
1486 if (input_dimension_product == output_dimension_product && in AlignLayouts()
1487 input_dimension_product > 1) { in AlignLayouts()
1489 input_dimension_product = output_dimension_product = 1; in AlignLayouts()
1491 if (input_dimension_product < output_dimension_product || in AlignLayouts()
1497 input_dimension_product *= input_shape.dimensions(i); in AlignLayouts()
1504 if (input_dimension_product != output_dimension_product) { in AlignLayouts()