Home
last modified time | relevance | path

Searched defs:FlattenGrad (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dflatten_grad.h31 FlattenGrad() : BaseOperator(kNameFlattenGrad) { InitIOName({"x", "shape"}, {"output"}); } in FlattenGrad() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
D_grad_ops.py477 class FlattenGrad(PrimitiveWithInfer): class