Searched refs:num_extended (Results 1 – 1 of 1) sorted by relevance
289 std::vector<int64> num_extended(num_spatial_dims); in ResizeUsingDilationAndConvolution() local294 num_extended[0] = upper_padding[0] / (dims.kernel_size[0]); in ResizeUsingDilationAndConvolution()295 num_extended[1] = upper_padding[1] / (dims.kernel_size[1]); in ResizeUsingDilationAndConvolution()299 if (num_extended[0] > 0) { in ResizeUsingDilationAndConvolution()303 for (int i = 0; i < num_extended[0]; i++) { in ResizeUsingDilationAndConvolution()308 if (num_extended[1] > 0) { in ResizeUsingDilationAndConvolution()311 {batch_dim_size, in_size[0] + num_extended[0], in_size[1], channels}, in ResizeUsingDilationAndConvolution()313 for (int i = 0; i < num_extended[1]; i++) { in ResizeUsingDilationAndConvolution()321 CalculateUpperPadding(in_size[0] + num_extended[0], out_size[0], in ResizeUsingDilationAndConvolution()324 CalculateUpperPadding(in_size[1] + num_extended[1], out_size[1], in ResizeUsingDilationAndConvolution()