warm_up.h 文件参考

warm-up function for torch model to avoid first multiple slowly inference 更多...

#include <vector>
#include "torch/extension.h"
#include "torch/script.h"

warm_up.h 的引用(Include)关系图:

此图展示该文件直接或间接的被哪些文件引用了:

浏览源代码.

命名空间
namespace	apollo::prediction
	apollo::prediction
namespace	apollo
	class register implement

函数
void	apollo::prediction::WarmUp (const std::vector< torch::jit::IValue > &torch_inputs, torch::jit::script::Module *model, at::Tensor *default_output)
	warm up function to avoid slowly inference of torch model

详细描述

warm-up function for torch model to avoid first multiple slowly inference

在文件 warm_up.h 中定义.