mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
379bbef23c
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/143355 Approved by: https://github.com/albanD
8 lines
189 B
C++
8 lines
189 B
C++
#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();
|