1 #pragma once 2 3 #include <ATen/native/DispatchStub.h> 4 #include <c10/util/ArrayRef.h> 5 6 /* 7 Depthwise 3x3 Winograd convolution operator 8 */ 9 10 namespace at { 11 class Tensor; 12 13 namespace native { 14 15 using convolution_depthwise3x3_winograd_fn = 16 Tensor (*)(const Tensor &, const Tensor &, const Tensor &, IntArrayRef, IntArrayRef, int64_t); 17 18 DECLARE_DISPATCH(convolution_depthwise3x3_winograd_fn, convolution_depthwise3x3_winograd_stub); 19 20 } // namespace native 21 } // namespace at 22