mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-08-21 08:14:00 +08:00
Harald Schäfer
f6885dcbec
* Revert "dmonitoringmodeld: use cl transform (#34235)" This reverts commit684b0b9d4d. * Revert "load model before calling convert_fp16_to_fp32" This reverts commit31606a7d15. * Revert "bump tinygrad" This reverts commit44f58ff758. * Revert "Tinygrad runner (#34171)" This reverts commit7b5a4fbb03. * Allow init buffer * typo
18 lines
443 B
C++
18 lines
443 B
C++
#pragma once
|
|
|
|
#include <string>
|
|
|
|
#include "selfdrive/modeld/runners/runmodel.h"
|
|
#include "selfdrive/modeld/thneed/thneed.h"
|
|
|
|
class ThneedModel : public RunModel {
|
|
public:
|
|
ThneedModel(const std::string path, float *_output, size_t _output_size, int runtime, bool use_tf8 = false, cl_context context = NULL);
|
|
void *getCLBuffer(const std::string name);
|
|
void execute();
|
|
private:
|
|
Thneed *thneed = NULL;
|
|
bool recorded;
|
|
float *output;
|
|
};
|