Home
last modified time | relevance | path

Searched defs:CreateTranspose (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dtranspose.c315 KernelBase *CreateTranspose(OpParameter *param, int data_type) { in CreateTranspose() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dconv2d_unify_mindir.cc79 CNodePtr CreateTranspose(const FuncGraphPtr &graph, const CNodePtr &conv2d, const AnfNodePtr &input… in CreateTranspose() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Ddynamic_rnn_grad_fission_v2.cc704 CNodePtr DynamicRnnGradFissionV2::CreateTranspose(const FuncGraphPtr &func_graph, const AnfNodePtr … in CreateTranspose() function in mindspore::opt::DynamicRnnGradFissionV2