File size: 197 Bytes
c1af2fa |
1 2 3 4 5 6 7 8 9 |
#pragma once
#include <c10/macros/Export.h>
TORCH_API int register_linear_params();
int register_embedding_params();
template <int kSpatialDim = 2> TORCH_API int register_conv_params();
|