forked from mindspore-Ecosystem/mindspore
!105 register not equal distributed op
Merge pull request !105 from lichen/register_not_equal_distributed_op
This commit is contained in:
commit
ad88c55859
|
@ -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();
|
||||
|
|
Loading…
Reference in New Issue