Caffe2 - C++ API
A deep learning, cross platform ML framework
python_tracer.h
1 #pragma once
2 
3 #include <torch/csrc/jit/tracer.h>
4 #include <torch/csrc/python_headers.h>
5 #include <torch/csrc/utils/pybind.h>
6 
7 #include <memory>
8 #include <string>
9 
10 namespace torch {
11 namespace jit {
12 namespace tracer {
13 void initPythonTracerBindings(PyObject* module);
14 
15 std::string getPythonInterpreterStackTrace();
16 Node* preRecordPythonTrace(
17  THPObjectPtr pyobj,
18  const std::string& arg_types,
20  pyobj_list scalar_args);
21 
22 std::shared_ptr<Graph> createGraphByTracing(
23  const py::function& func,
24  Stack inputs,
25  const py::function& var_name_lookup_fn,
26  bool force_outplace,
27  const c10::optional<size_t>& num_real_inputs = c10::nullopt);
28 } // namespace tracer
29 } // namespace jit
30 } // namespace torch
Definition: jit_type.h:17
ArrayRef - Represent a constant reference to an array (0 or more elements consecutively in memory)...
Definition: ArrayRef.h:41