|
@@ -1,4 +1,5 @@ |
|
|
#pragma once |
|
|
#pragma once |
|
|
|
|
|
#include <cstddef> |
|
|
#include "megdnn/arch.h" |
|
|
#include "megdnn/arch.h" |
|
|
#if MGB_ENABLE_DOT |
|
|
#if MGB_ENABLE_DOT |
|
|
|
|
|
|
|
@@ -21,4 +22,4 @@ void megdnn_dot_nchw_large_chanwise_direct_conv_11x11s2_oh4_ow16( |
|
|
const int8_t* src, const int8_t* weight, int32_t bias, int8_t* dst, size_t oh, |
|
|
const int8_t* src, const int8_t* weight, int32_t bias, int8_t* dst, size_t oh, |
|
|
size_t ow, size_t OH, size_t OW, size_t pad_iw, const float scale, |
|
|
size_t ow, size_t OH, size_t OW, size_t pad_iw, const float scale, |
|
|
int8_t relu_val); |
|
|
int8_t relu_val); |
|
|
#endif |
|
|
|
|
|
|
|
|
#endif |