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

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

muyangli's avatar
muyangli committed
6
7
// Tensor depthwise_conv2d_kernel(Tensor A, Tensor B);

Muyang Li's avatar
Muyang Li committed
8
Tensor dwconv_f16(Tensor input, Tensor weight, Tensor out, Tensor bias);