| TORCH_API int register_linear_params(); | |
| int register_embedding_params(); | |
| template <int kSpatialDim = 2> TORCH_API int register_conv_params(); | |
| TORCH_API int register_linear_params(); | |
| int register_embedding_params(); | |
| template <int kSpatialDim = 2> TORCH_API int register_conv_params(); | |