#include <pytorch.h>
Public Member Functions | |
void | Forward () |
void | ForwardCUDATensors () |
void | ForwardDynamic () |
Outputs & | GetOutputs () |
void | LoadModel (char *filename, int device) |
NeuralModel () | |
void | SetInputs (Inputs input) |
~NeuralModel () | |
Private Attributes | |
Inputs | _input |
Outputs | _output |
std::unique_ptr< NeuralModelImpl > | Model |
carla::learning::NeuralModel::NeuralModel | ( | ) |
Definition at line 159 of file pytorch.cpp.
carla::learning::NeuralModel::~NeuralModel | ( | ) |
Definition at line 290 of file pytorch.cpp.
void carla::learning::NeuralModel::Forward | ( | ) |
Definition at line 182 of file pytorch.cpp.
References carla::learning::GetWheelTensorInputs(), and carla::learning::GetWheelTensorOutput().
void carla::learning::NeuralModel::ForwardCUDATensors | ( | ) |
Definition at line 253 of file pytorch.cpp.
References carla::learning::GetWheelTensorOutput().
void carla::learning::NeuralModel::ForwardDynamic | ( | ) |
Definition at line 214 of file pytorch.cpp.
References carla::learning::GetWheelTensorInputs(), and carla::learning::GetWheelTensorOutputDynamic().
Outputs & carla::learning::NeuralModel::GetOutputs | ( | ) |
Definition at line 286 of file pytorch.cpp.
void carla::learning::NeuralModel::LoadModel | ( | char * | filename, |
int | device | ||
) |
Definition at line 162 of file pytorch.cpp.
void carla::learning::NeuralModel::SetInputs | ( | Inputs | input | ) |
Definition at line 177 of file pytorch.cpp.
|
private |