Module tch::jit

source ·
Expand description

JIT interface to run model trained/saved using PyTorch Python API.

Structs

CModule
A jit PyTorch module.
Object
TrainableCModule
The trainable version of a jit PyTorch module.

Enums

IValue
Argument and output values for JIT models.

Functions

f_get_profiling_mode
f_set_graph_executor_optimize
f_set_profiling_mode
get_profiling_mode
set_graph_executor_optimize
set_profiling_mode