p85620714
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!48803 Enable some operators's bprop expander. 

Merge pull request !48803 from TronZhang/enable_some_bprop_expand_part1
This commit is contained in:
i-robot 2023-02-17 01:23:20 +00:00 committed by Gitee
parent 14bec58e18 02675688a8
commit 7fd2a61163
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 45 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
mindspore/ccsrc/frontend/operator/graph_bprop/bprop_expander_meta_func_graph.cc
View File

@ -235,7 +235,51 @@ void RegNNBpropExpanderOps2() {
REGISTER_EXPANDER_BPROP_IMPL(UpsampleTrilinear3D);
}
void RegArrayBpropExpanderOps1() {}
void RegArrayBpropExpanderOps1() {
REGISTER_EXPANDER_BPROP_IMPL(Argmax);
REGISTER_EXPANDER_BPROP_IMPL(Argmin);
REGISTER_EXPANDER_BPROP_IMPL(BatchToSpace);
REGISTER_EXPANDER_BPROP_IMPL(CheckNumerics);
REGISTER_EXPANDER_BPROP_IMPL(Col2Im);
REGISTER_EXPANDER_BPROP_IMPL(ConjugateTranspose);
REGISTER_EXPANDER_BPROP_IMPL(DepthToSpace);
REGISTER_EXPANDER_BPROP_IMPL(Diag);
REGISTER_EXPANDER_BPROP_IMPL(DiagPart);
REGISTER_EXPANDER_BPROP_IMPL(DType);
REGISTER_EXPANDER_BPROP_IMPL(Fill);
REGISTER_EXPANDER_BPROP_IMPL(Fills);
REGISTER_EXPANDER_BPROP_IMPL(GatherD);
REGISTER_EXPANDER_BPROP_IMPL(Identity);
REGISTER_EXPANDER_BPROP_IMPL(IdentityN);
REGISTER_EXPANDER_BPROP_IMPL(MaskedSelect);
REGISTER_EXPANDER_BPROP_IMPL(MatrixDiagV3);
REGISTER_EXPANDER_BPROP_IMPL(NonZero);
REGISTER_EXPANDER_BPROP_IMPL(OnesLike);
REGISTER_EXPANDER_BPROP_IMPL(Range);
REGISTER_EXPANDER_BPROP_IMPL(Rank);
REGISTER_EXPANDER_BPROP_IMPL(ReverseSequence);
REGISTER_EXPANDER_BPROP_IMPL(ReverseV2);
REGISTER_EXPANDER_BPROP_IMPL(ScatterMax);
REGISTER_EXPANDER_BPROP_IMPL(ScatterMin);
REGISTER_EXPANDER_BPROP_IMPL(SegmentMax);
REGISTER_EXPANDER_BPROP_IMPL(SegmentMin);
REGISTER_EXPANDER_BPROP_IMPL(Select);
REGISTER_EXPANDER_BPROP_IMPL(Slice);
REGISTER_EXPANDER_BPROP_IMPL(SpaceToBatch);
REGISTER_EXPANDER_BPROP_IMPL(SpaceToDepth);
REGISTER_EXPANDER_BPROP_IMPL(Split);
REGISTER_EXPANDER_BPROP_IMPL(SplitV);
REGISTER_EXPANDER_BPROP_IMPL(TensorScatterAdd);
REGISTER_EXPANDER_BPROP_IMPL(TensorScatterDiv);
REGISTER_EXPANDER_BPROP_IMPL(TensorScatterElement);
REGISTER_EXPANDER_BPROP_IMPL(TensorScatterMul);
REGISTER_EXPANDER_BPROP_IMPL(TensorScatterSub);
REGISTER_EXPANDER_BPROP_IMPL(TensorScatterUpdate);
REGISTER_EXPANDER_BPROP_IMPL(Tril);
REGISTER_EXPANDER_BPROP_IMPL(Triu);
REGISTER_EXPANDER_BPROP_IMPL(ZerosLike);
}
void RegArrayBpropExpanderOps2() {}
void RegClipBpropExpanderOps() {}
void RegCommBpropExpanderOps() {}