Searched refs:original_output_shape (Results 1 – 1 of 1) sorted by relevance
1085 const auto& original_output_shape = output.shape(); in CalculateOutputSize() local1087 int num_dims = std::max(1, original_output_shape.dim_size()); in CalculateOutputSize()1088 auto output_shape = MaybeGetMinimumShape(original_output_shape, num_dims, in CalculateOutputSize()