Searched defs:TensorRTPlugin (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | vsl_compress_tensorrt.h | 30 : TensorRTPlugin(name, std::string(VSL_COMPRESS_PLUGIN_NAME), device_id), in TensorRTPlugin() function
33 : TensorRTPlugin(std::string(name), std::string(VSL_COMPRESS_PLUGIN_NAME)), in TensorRTPlugin() function
37 : TensorRTPlugin(std::string(name), std::string(VSL_COMPRESS_PLUGIN_NAME)), in TensorRTPlugin() function
|
| D | cast_plugin.h | 27 … : TensorRTPlugin(name, std::string(CAST_PLUGIN_NAME), device_id), dest_datatype_(dest_datatype) {} in TensorRTPlugin() function
|