!105 register not equal distributed op

Merge pull request !105 from lichen/register_not_equal_distributed_op
This commit is contained in:
mindspore-ci-bot 2020-04-02 21:34:34 +08:00 committed by Gitee
commit ad88c55859
3 changed files with 3 additions and 1 deletions

View File

@ -180,6 +180,7 @@ constexpr char SIGMOID[] = "Sigmoid";
constexpr char POW[] = "Pow";
constexpr char MAXIMUM[] = "Maximum";
constexpr char EQUAL[] = "Equal";
constexpr char NOT_EQUAL[] = "NotEqual";
constexpr char LOGICALNOT[] = "LogicalNot";
constexpr char GATHERV2[] = "GatherV2";
constexpr char STRIDEDSLICE[] = "StridedSlice";

View File

@ -95,6 +95,7 @@ std::vector<std::string> splittable_op_ = {MATMUL,
POW,
MAXIMUM,
EQUAL,
NOT_EQUAL,
LOGICALNOT,
GATHERV2,
STRIDEDSLICE,

View File

@ -1226,7 +1226,7 @@ void CoverSliceShape(const FuncGraphPtr& root) {
MS_LOG(INFO) << "Parameter " << parameter->ToString() << " don't need to set parallel shape";
} else {
SetParallelShape(parameter, res);
MS_LOG(DEBUG) << "Parameter " << parameter->ToString() << " shape " << parameter->Shape()->ToString();
MS_LOG(DEBUG) << "Parameter " << parameter->ToString() << " shape " << parameter->Shape()->ToString();
}
}
g_RefMap.clear();