Support TensorRT provider (#921)
Signed-off-by: manickavela1998@gmail.com <manickavela1998@gmail.com> Signed-off-by: manickavela1998@gmail.com <manickavela.arumugam@uniphore.com>
This commit is contained in:
@@ -24,6 +24,8 @@ Provider StringToProvider(std::string s) {
|
||||
return Provider::kXnnpack;
|
||||
} else if (s == "nnapi") {
|
||||
return Provider::kNNAPI;
|
||||
} else if (s == "trt") {
|
||||
return Provider::kTRT;
|
||||
} else {
|
||||
SHERPA_ONNX_LOGE("Unsupported string: %s. Fallback to cpu", s.c_str());
|
||||
return Provider::kCPU;
|
||||
|
||||
Reference in New Issue
Block a user