#include <pytorch.h>
Collaboration diagram for carla::learning::NeuralModel: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().
Here is the call graph for this function:| void carla::learning::NeuralModel::ForwardCUDATensors | ( | ) |
Definition at line 253 of file pytorch.cpp.
References carla::learning::GetWheelTensorOutput().
Here is the call graph for this function:| void carla::learning::NeuralModel::ForwardDynamic | ( | ) |
Definition at line 214 of file pytorch.cpp.
References carla::learning::GetWheelTensorInputs(), and carla::learning::GetWheelTensorOutputDynamic().
Here is the call graph for this function:| 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 |
1.8.13