#pragma once #include "common.h" #include "Tensor.h" Tensor depthwise_conv2d_kernel(Tensor A, Tensor B);