Function deepmd::create_dipole_charge_modifier_backend_from_plugin

Function deepmd::create_dipole_charge_modifier_backend_from_plugin#

Function Documentation#

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

Create a DipoleChargeModifier 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 DipoleChargeModifier backend instance.