FM / backends

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

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

Name	Size	Mode	Actions
backend.h	4060	0644	edit dl rm
backend_debug_handler.h	6332	0644	edit dl rm
backend_debug_info.h	2338	0644	edit dl rm
backend_detail.h	1104	0644	edit dl rm
backend_exception.h	2052	0644	edit dl rm
backend_init.h	277	0644	edit dl rm
backend_interface.h	1184	0644	edit dl rm
backend_preprocess.h	438	0644	edit dl rm
backend_resolver.h	277	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/backends/backend_detail.h (1104B)

#pragma once

#include <torch/csrc/jit/api/module.h>

#include <ATen/core/jit_type.h>

#include <functional>

namespace torch {
namespace jit {

using DebugHandleType = int64_t;

using NodeToDebugHandle = std::unordered_map<Node*, DebugHandleType>;

using BackendDebugHandleGenerator =
    std::function<NodeToDebugHandle(const std::shared_ptr<Graph>&)>;

namespace detail {

using BackendPreprocessFunction = std::function<c10::IValue(
    const Module&,
    const c10::Dict<IValue, IValue>&,
    const BackendDebugHandleGenerator& generate_debug_handles)>;

TORCH_API void registerBackendPreprocessFunction(
    const std::string& name,
    const BackendPreprocessFunction& preprocess);

bool hasBackendPreprocessFunction(const std::string& name);

BackendPreprocessFunction getBackendPreprocessFunction(const std::string& name);

TORCH_API Module codegen_backend_module(
    const std::string& backend_name,
    const Module& orig_module,
    const c10::Dict<IValue, IValue>& method_compile_spec,
    const c10::DictTypePtr& any_dict_ty);
} // namespace detail
} // namespace jit
} // namespace torch

#pragma once #include #include #include namespace torch { namespace jit { using DebugHandleType = int64_t; using NodeToDebugHandle = std::unordered_map; using BackendDebugHandleGenerator = std::function&)>; namespace detail { using BackendPreprocessFunction = std::function&, const BackendDebugHandleGenerator& generate_debug_handles)>; TORCH_API void registerBackendPreprocessFunction( const std::string& name, const BackendPreprocessFunction& preprocess); bool hasBackendPreprocessFunction(const std::string& name); BackendPreprocessFunction getBackendPreprocessFunction(const std::string& name); TORCH_API Module codegen_backend_module( const std::string& backend_name, const Module& orig_module, const c10::Dict& method_compile_spec, const c10::DictTypePtr& any_dict_ty); } // namespace detail } // namespace jit } // namespace torch