FM / runtime

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

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

Name	Size	Mode	Actions
argument_spec.h	16896	0644	edit dl rm
autodiff.h	3993	0644	edit dl rm
calculate_necessary_args.h	2100	0644	edit dl rm
custom_operator.h	1079	0644	edit dl rm
exception_message.h	639	0644	edit dl rm
graph_executor.h	4477	0644	edit dl rm
graph_executor_impl.h	3902	0644	edit dl rm
graph_iterator.h	4963	0644	edit dl rm
instruction.h	3894	0644	edit dl rm
interpreter.h	5081	0644	edit dl rm
jit_exception.h	270	0644	edit dl rm
logging.h	2618	0644	edit dl rm
operator.h	9060	0644	edit dl rm
operator_options.h	189	0644	edit dl rm
print_handler.h	340	0644	edit dl rm
profiling_graph_executor_impl.h	2036	0644	edit dl rm
profiling_record.h	8962	0644	edit dl rm
register_ops_utils.h	43794	0644	edit dl rm
script_profile.h	2532	0644	edit dl rm
slice_indices_adjust.h	821	0644	edit dl rm
symbolic_script.h	598	0644	edit dl rm
symbolic_shape_registry.h	425	0644	edit dl rm
vararg_functions.h	1023	0644	edit dl rm
variable_tensor_list.h	545	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/runtime/custom_operator.h (1079B)

#pragma once

#include <ATen/core/op_registration/op_registration.h>
#include <ATen/core/stack.h>
#include <torch/csrc/jit/runtime/operator.h>

namespace torch {
namespace jit {

/// Registration class for new operators. Effectively calls
/// `torch::jit::registerOperator` for every supplied operator, but allows doing
/// so in the global scope when a `RegisterOperators` object is assigned to a
/// static variable.
/// Note: This is *not* the custom operator API. If you want to register custom
/// operators, take a look at torch::RegisterOperators.
struct TORCH_API RegisterOperators {
  RegisterOperators() = default;

/// Registers a vector of already created `Operator`s.
  /// The operator element is now optional to filter null ops. It's backward
  /// compatible and works for selective operator registration.
  explicit RegisterOperators(std::vector<c10::optional<Operator>> operators) {
    for (c10::optional<Operator>& o : operators) {
      if (o) {
        registerOperator(std::move(o.value()));
      }
    }
  }
};

} // namespace jit
} // namespace torch

#pragma once #include #include #include namespace torch { namespace jit { /// Registration class for new operators. Effectively calls /// `torch::jit::registerOperator` for every supplied operator, but allows doing /// so in the global scope when a `RegisterOperators` object is assigned to a /// static variable. /// Note: This is *not* the custom operator API. If you want to register custom /// operators, take a look at torch::RegisterOperators. struct TORCH_API RegisterOperators { RegisterOperators() = default; /// Registers a vector of already created `Operator`s. /// The operator element is now optional to filter null ops. It's backward /// compatible and works for selective operator registration. explicit RegisterOperators(std::vector> operators) { for (c10::optional& o : operators) { if (o) { registerOperator(std::move(o.value())); } } } }; } // namespace jit } // namespace torch