Home
last modified time | relevance | path

Searched defs:GetCustomShape (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/providers/nnie/src/
Dcustom_infer.cc41 int GetCustomShape(const mindspore::schema::Custom *op, const std::string &attr, in GetCustomShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/providers/dpico/infer/
Dcustom_infer.cc40 Status GetCustomShape(const std::map<std::string, std::string> &attrs, const std::string &attr_name… in GetCustomShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/common/
Dutils.cc42 int GetCustomShape(const std::string &attr, std::vector<std::vector<int>> *shapes) { in GetCustomShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/infer/
Ddpico_custom_infer.cc77 Status GetCustomShape(const std::map<std::string, std::string> &attrs, const std::string &attr_name… in GetCustomShape() function