Searched defs:max_dim_size (Results 1 – 2 of 2) sorted by relevance
122 auto max_dim_size = output_type.getDimSize(dimension); in matchAndRewrite() local
611 const int max_dim_size = slice_shape.dim_size(i) * num_concats_[i]; in GetInputsAndOutputShape() local