Searched refs:original_output_shape (Results 1 – 1 of 1) sorted by relevance
1593 const auto& original_output_shape = output.shape(); in CalculateOutputSize() local1595 int num_dims = std::max(1, original_output_shape.dim_size()); in CalculateOutputSize()1596 auto output_shape = MaybeGetMinimumShape(original_output_shape, num_dims, in CalculateOutputSize()1615 const auto& original_output_shape = output.shape(); in CalculateOutputTensorSize() local1617 int num_dims = std::max(1, original_output_shape.dim_size()); in CalculateOutputTensorSize()1618 auto output_shape = MaybeGetMinimumShape(original_output_shape, num_dims, in CalculateOutputTensorSize()