#include "pytorch.h"
#include <torch/torch.h>
#include <torch/script.h>
#include <torchscatter/scatter.h>
#include <torchcluster/cluster.h>
#include <torch/csrc/jit/passes/tensorexpr_fuser.h>
#include <c10/cuda/CUDACachingAllocator.h>
#include <string>
#include <vector>
#include <ostream>
#include <sstream>
Go to the source code of this file.
Classes | |
struct | carla::learning::NeuralModelImpl |
Namespaces | |
carla | |
This file contains definitions of common data structures used in traffic manager. | |
carla::learning | |
Macros | |
#define | _GLIBCXX_USE_CXX11_ABI 0 |
Functions | |
void | add_mark (const std::string &text) |
torch::jit::IValue | carla::learning::GetWheelTensorInputs (WheelInput &wheel) |
WheelOutput | carla::learning::GetWheelTensorOutput (const at::Tensor &particle_forces, const at::Tensor &wheel_forces) |
WheelOutput | carla::learning::GetWheelTensorOutputDynamic (const at::Tensor &particle_forces, const at::Tensor &wheel_forces) |
void | carla::learning::test_learning () |
#define _GLIBCXX_USE_CXX11_ABI 0 |
Definition at line 5 of file pytorch.cpp.
void add_mark | ( | const std::string & | text | ) |
Definition at line 20 of file pytorch.cpp.