FM / frontend

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/frontend/

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/frontend

Name	Size	Mode	Actions
builtin_functions.h	254	0644	edit dl rm
canonicalize_modified_loop.h	326	0644	edit dl rm
code_template.h	7057	0644	edit dl rm
concrete_module_type.h	9040	0644	edit dl rm
convert_to_ssa.h	341	0644	edit dl rm
edit_distance.h	268	0644	edit dl rm
error_report.h	1480	0644	edit dl rm
exit_transforms.h	232	0644	edit dl rm
function_schema_parser.h	460	0644	edit dl rm
inline_loop_condition.h	366	0644	edit dl rm
ir_emitter.h	541	0644	edit dl rm
lexer.h	18470	0644	edit dl rm
mini_environment.h	1366	0644	edit dl rm
name_mangler.h	669	0644	edit dl rm
parser.h	688	0644	edit dl rm
parser_constants.h	162	0644	edit dl rm
parse_string_literal.h	2295	0644	edit dl rm
resolver.h	1972	0644	edit dl rm
schema_matching.h	1948	0644	edit dl rm
schema_type_parser.h	977	0644	edit dl rm
script_type_parser.h	1619	0644	edit dl rm
source_range.h	6038	0644	edit dl rm
source_ref.h	1269	0644	edit dl rm
strtod.h	240	0644	edit dl rm
sugared_value.h	26805	0644	edit dl rm
tracer.h	11925	0644	edit dl rm
tree.h	6635	0644	edit dl rm
tree_views.h	35696	0644	edit dl rm
versioned_symbols.h	597	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/frontend/schema_matching.h (1948B)

#pragma once
#include <torch/csrc/WindowsTorchApiMacro.h>
#include <torch/csrc/jit/ir/ir.h>
#include <torch/csrc/jit/ir/named_value.h>

#include <ATen/core/function_schema.h>

namespace torch {
namespace jit {

// Try to match a list of inputs and keyword 'attributes' to this
// schema. Return the flat list of positional inputs to the call or
// `c10::nullopt` on failure (`failure_messages` contains a good error
// report in this case)

struct MatchedSchema {
  std::vector<Value*> inputs;
  std::vector<TypePtr> return_types;
  c10::OptNameList return_field_names;
};

TORCH_API MatchedSchema matchSchema(
    const ::c10::FunctionSchema& schema,
    const SourceRange& loc,
    Graph& graph,
    at::ArrayRef<NamedValue> args,
    at::ArrayRef<NamedValue> kwargs,
    const c10::optional<NamedValue>& self = c10::nullopt);

TORCH_API std::pair<size_t, MatchedSchema> matchSchemas(
    const std::vector<const ::c10::FunctionSchema*>& schemas,
    const SourceRange& loc,
    Graph& graph,
    at::ArrayRef<NamedValue> args,
    at::ArrayRef<NamedValue> kwargs,
    const c10::optional<NamedValue>& self = c10::nullopt,
    bool render_errors = false);

TORCH_API bool convertibleToList(
    const TypePtr& type,
    const TypePtr& list_type_);

TORCH_API Value* emitBuiltinCall(
    const SourceRange& loc,
    Graph& graph,
    Symbol name,
    at::ArrayRef<NamedValue> args,
    at::ArrayRef<NamedValue> kwargs,
    const c10::optional<NamedValue>& self = c10::nullopt);

TORCH_API c10::optional<size_t> findInputWithName(
    const std::string& name,
    at::ArrayRef<NamedValue> kwargs);

// applies implicit conversion from value trying to turn it into type
// concrete_type it succeeds if the return_value->isSubtypeOf(concrete_type)
TORCH_API Value* tryConvertToType(
    const SourceRange& loc,
    Graph& graph,
    const TypePtr& concrete_type,
    Value* value,
    bool allow_conversions);
} // namespace jit
} // namespace torch

#pragma once #include #include #include #include namespace torch { namespace jit { // Try to match a list of inputs and keyword 'attributes' to this // schema. Return the flat list of positional inputs to the call or // `c10::nullopt` on failure (`failure_messages` contains a good error // report in this case) struct MatchedSchema { std::vector inputs; std::vector return_types; c10::OptNameList return_field_names; }; TORCH_API MatchedSchema matchSchema( const ::c10::FunctionSchema& schema, const SourceRange& loc, Graph& graph, at::ArrayRef args, at::ArrayRef kwargs, const c10::optional& self = c10::nullopt); TORCH_API std::pair matchSchemas( const std::vector& schemas, const SourceRange& loc, Graph& graph, at::ArrayRef args, at::ArrayRef kwargs, const c10::optional& self = c10::nullopt, bool render_errors = false); TORCH_API bool convertibleToList( const TypePtr& type, const TypePtr& list_type_); TORCH_API Value* emitBuiltinCall( const SourceRange& loc, Graph& graph, Symbol name, at::ArrayRef args, at::ArrayRef kwargs, const c10::optional& self = c10::nullopt); TORCH_API c10::optional findInputWithName( const std::string& name, at::ArrayRef kwargs); // applies implicit conversion from value trying to turn it into type // concrete_type it succeeds if the return_value->isSubtypeOf(concrete_type) TORCH_API Value* tryConvertToType( const SourceRange& loc, Graph& graph, const TypePtr& concrete_type, Value* value, bool allow_conversions); } // namespace jit } // namespace torch