Searched defs:kernel_ndim (Results 1 – 1 of 1) sorted by relevance
64 bool kernel_size_is_valid(IntArrayRef kernel_size, size_t kernel_ndim) { in kernel_size_is_valid()73 bool stride_is_valid(IntArrayRef stride, size_t kernel_ndim, bool allow_empty) { in stride_is_valid()81 size_t kernel_ndim, in padding_is_valid()108 bool dilation_is_valid(IntArrayRef dilation, size_t kernel_ndim) { in dilation_is_valid()117 size_t kernel_ndim) { in output_padding_is_valid()138 size_t kernel_ndim) { in output_size_is_valid()225 size_t kernel_ndim, in calculate_kernel_output_sizes()354 size_t kernel_ndim = 2; in check_convolution_args() local430 size_t kernel_ndim = 2; in get_convolution_out_target_size() local