提交 10312620 编写于 作者: L lichenever 提交者: 高东海

register not equal distributed op

上级 65cd9d48
......@@ -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";
......
......@@ -95,6 +95,7 @@ std::vector<std::string> splittable_op_ = {MATMUL,
POW,
MAXIMUM,
EQUAL,
NOT_EQUAL,
LOGICALNOT,
GATHERV2,
STRIDEDSLICE,
......
......@@ -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();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册