Home
last modified time | relevance | path

Searched defs:GetGPUGroupSize (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/distribution/
Ddistribution_base.cc21 int GetGPUGroupSize() { return TensorRTExecutorPlugin::GetInstance().GetGPUGroupSize(); } in GetGPUGroupSize() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_plugin_impl.cc23 int TensorRTPluginImpl::GetGPUGroupSize() const { in GetGPUGroupSize() function in mindspore::lite::TensorRTPluginImpl
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/plugin/
Dtensorrt_executor_plugin.cc82 int TensorRTExecutorPlugin::GetGPUGroupSize() { in GetGPUGroupSize() function in mindspore::lite::TensorRTExecutorPlugin