forked from mindspore-Ecosystem/mindspore
fix api doc
This commit is contained in:
parent
02f58183c5
commit
b16544aba1
|
@ -101,7 +101,7 @@ class MS_API Context {
|
|||
|
||||
/// \brief Set Delegate to access third-party AI framework. Only valid for Lite.
|
||||
///
|
||||
/// \param[in] Pointer to the custom delegate.
|
||||
/// \param[in] delegate the custom delegate.
|
||||
void SetDelegate(const std::shared_ptr<Delegate> &delegate);
|
||||
|
||||
/// \brief Get the delegate of the third-party AI framework. Only valid for Lite.
|
||||
|
@ -145,7 +145,6 @@ class MS_API DeviceInfoContext : public std::enable_shared_from_this<DeviceInfoC
|
|||
/// \brief A similar function to RTTI is provided when the -fno-rtti compilation option is turned on, which converts
|
||||
/// DeviceInfoContext to a shared pointer of type T, and returns nullptr if the conversion fails.
|
||||
///
|
||||
/// \param T Type
|
||||
/// \return A pointer of type T after conversion. If the conversion fails, it will be nullptr.
|
||||
template <class T>
|
||||
std::shared_ptr<T> Cast() {
|
||||
|
@ -393,7 +392,7 @@ class MS_API AscendDeviceInfo : public DeviceInfoContext {
|
|||
|
||||
/// \brief Set the dynamic image size of model inputs.
|
||||
///
|
||||
/// \param[in] image size hw e.g. "66,88;32,64" means h1:66,w1:88; h2:32,w2:64.
|
||||
/// \param[in] dynamic_image_size size hw e.g. "66,88;32,64" means h1:66,w1:88; h2:32,w2:64.
|
||||
inline void SetDynamicImageSize(const std::string &dynamic_image_size);
|
||||
|
||||
/// \brief Get dynamic image size of model inputs.
|
||||
|
|
|
@ -124,7 +124,7 @@ class MS_API Kernel {
|
|||
}
|
||||
/// \brief set kernel's config.
|
||||
///
|
||||
/// \param[in] config define the kernel's config.
|
||||
/// \param[in] section define the section of the kernel's config.
|
||||
std::map<std::string, std::string> GetConfig(const std::string §ion) const {
|
||||
if (config_ == nullptr) {
|
||||
return std::map<std::string, std::string>();
|
||||
|
|
|
@ -96,6 +96,7 @@ class MS_API NodeParser {
|
|||
/// \brief Method to parse node of TFLITE
|
||||
///
|
||||
/// \param[in] tflite_op Define the node to be resolved.
|
||||
/// \param[in] tflite_subgraph Define the subgraph to be resolved.
|
||||
/// \param[in] tflite_model Define the model, which contains all information abort the graph.
|
||||
///
|
||||
/// \return PrimitiveC Attribute storage.
|
||||
|
|
Loading…
Reference in New Issue