Module tch::jit[][src]

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

Structs

CModule

A jit PyTorch module.

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_profiling_mode
get_profiling_mode
set_profiling_mode