/usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/cpu
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/cpu/DepthwiseConvKernel.h (447B)
#pragma once
#include
#include
/*
Depthwise 3x3 Winograd convolution operator
*/
namespace at {
namespace native {
using convolution_depthwise3x3_winograd_fn =
Tensor (*)(const Tensor &, const Tensor &, const Tensor &,IntArrayRef, IntArrayRef, int64_t);
DECLARE_DISPATCH(convolution_depthwise3x3_winograd_fn, convolution_depthwise3x3_winograd_stub);
} // namespace native
} // namespace at