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_interface.h (1184B)

#pragma once

#include <torch/custom_class.h>

namespace torch {
namespace jit {

// Interface for a JIT backend.
class TORCH_API PyTorchBackendInterface : public torch::CustomClassHolder {
 public:
  PyTorchBackendInterface() noexcept;
  ~PyTorchBackendInterface() override;

// Returns true if the backend is available to process delegation calls.
  virtual bool is_available() = 0;

// Compile the module contained in \p processed using the details provided in
  // \p method_compile_spec for each module method that should be compiled for
  // the backend. \p method_compile_spec should be of type Dict<string, Any>.
  // \returns a dictionary of type Dict<string, Any> that contains a backend
  // handle each method that can run on the backend (i.e. each key in \p
  // method_compile_spec).
  virtual c10::impl::GenericDict compile(
      c10::IValue processed,
      c10::impl::GenericDict method_compile_spec) = 0;

// Execute the method specified by \p handle using \p inputs. \returns the
  // outputs as a tuple.
  virtual c10::impl::GenericList execute(
      c10::IValue handle,
      c10::impl::GenericList inputs) = 0;
};
} // namespace jit
} // namespace torch

#pragma once #include namespace torch { namespace jit { // Interface for a JIT backend. class TORCH_API PyTorchBackendInterface : public torch::CustomClassHolder { public: PyTorchBackendInterface() noexcept; ~PyTorchBackendInterface() override; // Returns true if the backend is available to process delegation calls. virtual bool is_available() = 0; // Compile the module contained in \p processed using the details provided in // \p method_compile_spec for each module method that should be compiled for // the backend. \p method_compile_spec should be of type Dict. // \returns a dictionary of type Dict that contains a backend // handle each method that can run on the backend (i.e. each key in \p // method_compile_spec). virtual c10::impl::GenericDict compile( c10::IValue processed, c10::impl::GenericDict method_compile_spec) = 0; // Execute the method specified by \p handle using \p inputs. \returns the // outputs as a tuple. virtual c10::impl::GenericList execute( c10::IValue handle, c10::impl::GenericList inputs) = 0; }; } // namespace jit } // namespace torch