Browse Source

!25843 [ME][Fallback] Add Primitive type into ValueNameToConverterVector

Merge pull request !25843 from Margaret_wangrui/fallback_bug
tags/v1.6.0
i-robot Gitee 4 years ago
parent
commit
49d38f84e1
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      mindspore/ccsrc/utils/convert_utils_py.cc

+ 3
- 1
mindspore/ccsrc/utils/convert_utils_py.cc View File

@@ -219,7 +219,9 @@ static ValueNameToConverterVector value_name_to_converter = {
// Monad
{Monad::kTypeId, [](const ValuePtr &value) -> py::object { return py::none(); }},
// Ellipsis
{Ellipsis::kTypeId, [](const ValuePtr &value) -> py::object { return py::ellipsis(); }}};
{Ellipsis::kTypeId, [](const ValuePtr &value) -> py::object { return py::ellipsis(); }},
// Primitive
{Primitive::kTypeId, [](const ValuePtr &value) -> py::object { return py::none(); }}};

py::object ValueToPyData(const ValuePtr &value) {
if (value == nullptr) {


Loading…
Cancel
Save