Files
pytorch/torch/csrc/utils/tensor_qschemes.h
2024-06-04 15:12:45 +00:00

10 lines
174 B
C++

#pragma once
#include <torch/csrc/QScheme.h>
namespace torch::utils {
PyObject* getTHPQScheme(at::QScheme qscheme);
void initializeQSchemes();
} // namespace torch::utils