dwconv.h 106 Bytes
Newer Older
Zhekai Zhang's avatar
Zhekai Zhang committed
1
2
3
4
5
6
#pragma once

#include "common.h"
#include "Tensor.h"

Tensor depthwise_conv2d_kernel(Tensor A, Tensor B);