Searched refs:input_cl_check (Results 1 – 1 of 1) sorted by relevance
634 Tensor input_cl_check = input.unsqueeze(0); in max_pool3d_kernel_impl() local639 if ((!input_cl_check.is_contiguous()) && in max_pool3d_kernel_impl()640 input_cl_check.is_contiguous(at::MemoryFormat::ChannelsLast3d)) { in max_pool3d_kernel_impl()649 cpu_max_pool_channels_last<scalar_t, /*is 3d*/true>(output, indices, input_cl_check, in max_pool3d_kernel_impl()