script_type_parser.h

#pragma once
#include <ATen/core/jit_type.h>
#include <torch/csrc/WindowsTorchApiMacro.h>
#include <torch/csrc/jit/script/parser.h>

namespace torch {
namespace jit {
namespace script {
struct Expr;
TORCH_API c10::optional<std::string> parseBaseTypeName(const Expr& expr);
TORCH_API c10::TypePtr parseTypeFromExpr(const Expr& expr);
TORCH_API c10::optional<std::pair<c10::TypePtr, int32_t>> parseBroadcastList(
    const Expr& expr);
TORCH_API c10::TypePtr parseType(const std::string& str);
} // namespace script
} // namespace jit
} // namespace torch