1 #pragma once 2 3 #include <torch/csrc/Export.h> 4 #include <torch/csrc/jit/ir/ir.h> 5 6 namespace torch::jit { 7 8 TORCH_API const std::string& GetSerializedShapeFunctions(); 9 10 TORCH_API const OperatorMap<std::string>& GetShapeFunctionMappings(); 11 12 TORCH_API const OperatorMap<std::pair<std::string, std::string>>& 13 GetBoundedShapeMappings(); 14 15 } // namespace torch::jit 16