Searched defs:GetVmapRuleFunction (Results 1 – 2 of 2) sorted by relevance
117 py::function GetVmapRuleFunction(const std::string &name, int axis_size) { in GetVmapRuleFunction() function
178 py::function PrimitivePy::GetVmapRuleFunction(const bool, int axis_size) { in GetVmapRuleFunction() function in mindspore::PrimitivePy