FM / python

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

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

Name	Size	Mode	Actions
init.h	151	0644	edit dl rm
module_python.h	685	0644	edit dl rm
pybind.h	8102	0644	edit dl rm
pybind_utils.h	39711	0644	edit dl rm
python_arg_flatten.h	3577	0644	edit dl rm
python_custom_class.h	471	0644	edit dl rm
python_dict.h	3345	0644	edit dl rm
python_ir.h	1718	0644	edit dl rm
python_ivalue.h	3249	0644	edit dl rm
python_list.h	5438	0644	edit dl rm
python_sugared_value.h	10545	0644	edit dl rm
python_tracer.h	911	0644	edit dl rm
python_tree_views.h	175	0644	edit dl rm
script_init.h	177	0644	edit dl rm
update_graph_executor_opt.h	224	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/python/python_ir.h (1718B)

#pragma once

#include <torch/csrc/jit/ir/ir.h>
#include <torch/csrc/utils/object_ptr.h>

namespace torch {
namespace jit {

void initPythonIRBindings(PyObject* module);

// execute a Python function, used for Ops we can't optimize but that we want to
// optimize around
struct ConcretePythonOp : public PythonOp {
  static Symbol Kind;

ConcretePythonOp(Graph* graph) : PythonOp(graph, ::c10::prim::PythonOp) {}
  ConcretePythonOp* init(
      THPObjectPtr&& pyobj,
      const std::string& cconv,
      pyobj_list&& scalar_args) {
    this->pyobj = std::move(pyobj);
    this->scalar_args = std::move(scalar_args);
    this->cconv = cconv;
    return this;
  }
  // The Python object which contains the implementation of this function.
  // This is either a class (non-legacy) or an object (legacy).  See
  // TraceInterpreterState for execution semantics.
  THPObjectPtr pyobj;
  // The calling convention for the Python function.
  // 'c' -- constant argument
  // 'd' -- dynamic argument
  std::string cconv;
  // Scalar arguments to the Python function.  Not necessarily passed to
  // the function in this order; see cconv for the correct order.
  std::vector<THPObjectPtr> scalar_args;

std::string name() const override;
  void cloneFrom(Node* other_) override;
  Node* allocNewInstance(Graph* g) override {
    return new ConcretePythonOp(g);
  }
  // recover the autograd.Function instance, if this PythonOp's function
  // was originally SomeFunction.apply
  // used in ONNX for discovering symbolics
  c10::optional<THPObjectPtr> autogradFunction() const override;
  void writeScalars(std::ostream& out) const override;
  void lint_python() const override;
};

} // namespace jit
} // namespace torch

#pragma once #include #include namespace torch { namespace jit { void initPythonIRBindings(PyObject* module); // execute a Python function, used for Ops we can't optimize but that we want to // optimize around struct ConcretePythonOp : public PythonOp { static Symbol Kind; ConcretePythonOp(Graph* graph) : PythonOp(graph, ::c10::prim::PythonOp) {} ConcretePythonOp* init( THPObjectPtr&& pyobj, const std::string& cconv, pyobj_list&& scalar_args) { this->pyobj = std::move(pyobj); this->scalar_args = std::move(scalar_args); this->cconv = cconv; return this; } // The Python object which contains the implementation of this function. // This is either a class (non-legacy) or an object (legacy). See // TraceInterpreterState for execution semantics. THPObjectPtr pyobj; // The calling convention for the Python function. // 'c' -- constant argument // 'd' -- dynamic argument std::string cconv; // Scalar arguments to the Python function. Not necessarily passed to // the function in this order; see cconv for the correct order. std::vector scalar_args; std::string name() const override; void cloneFrom(Node* other_) override; Node* allocNewInstance(Graph* g) override { return new ConcretePythonOp(g); } // recover the autograd.Function instance, if this PythonOp's function // was originally SomeFunction.apply // used in ONNX for discovering symbolics c10::optional autogradFunction() const override; void writeScalars(std::ostream& out) const override; void lint_python() const override; }; } // namespace jit } // namespace torch