Module tch::jit

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_get_tensor_expr_fuser_enabled
• f_set_graph_executor_optimize
  Enables or disables the graph executor optimizer for the current thread.
• f_set_profiling_mode
• f_set_tensor_expr_fuser_enabled
• get_profiling_mode
• get_tensor_expr_fuser_enabled
• set_graph_executor_optimize
  Enables or disables the graph executor optimizer for the current thread.
• set_profiling_mode
• set_tensor_expr_fuser_enabled