Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc487 auto output_shape_at_dim = in ValidateAndGetTiledExecuteOutputShape() local
489 if (output_shape_at_dim % output_splits != 0) { in ValidateAndGetTiledExecuteOutputShape()
496 dimension_index, output_splits, output_shape_at_dim)); in ValidateAndGetTiledExecuteOutputShape()