Function deepmd::create_deeptensor_backend_from_plugin

Function deepmd::create_deeptensor_backend_from_plugin#

Function Documentation#

std::shared_ptr<DeepTensorBase> deepmd::create_deeptensor_backend_from_plugin(DPBackend backend, const std::string &model, const int &gpu_rank, const std::string &name_scope)#

Create a DeepTensor backend through a runtime backend plugin.

Parameters:

  • backend – [in] The backend plugin to load.

  • model – [in] The name of the frozen model file.

  • gpu_rank – [in] The GPU rank.

  • name_scope – [in] The TensorFlow name scope.

Returns:

The DeepTensor backend instance.