Searched defs:trans_op (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/delegate/npu/pass/ |
| D | npu_pass_utils.cc | 60 void NPUPassUtils::UpdateNH2NCTransNodePreOp(NPUOp *pre_op, NPUOp *trans_op, NPUOp *op) { in UpdateNH2NCTransNodePreOp()
93 void NPUPassUtils::UpdateNH2NCTransNodePostOp(NPUOp *trans_op, NPUOp *post_op) { in UpdateNH2NCTransNodePostOp()
100 void NPUPassUtils::UpdateNC2NHPostOpInTensors(NPUOp *op, NPUOp *trans_op, NPUOp *post_op) { in UpdateNC2NHPostOpInTensors()
113 void NPUPassUtils::UpdateNC2NHTransNodePostOp(NPUOp *op, NPUOp *trans_op, NPUOp *post_op) { in UpdateNC2NHTransNodePostOp()
|
| D | npu_transform_pass.cc | 60 auto *trans_op = NPUPassUtils::CreateNhwc2NchwOp({op->inputs()[0]}, pre_trans_outputs, name); in InsertPreNodes() local
|
| D | npu_fusion_pass.cc | 381 for (const auto &trans_op : cur_op->out_ops()) { in FormatFusion() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | transform_op_optimizer.cc | 344 auto trans_op = op_checker_.GenTransformOp(trans_type); in RebuildLiteGraph() local
359 auto &trans_op = trans_op_cache[node_id_from]; in RebuildLiteGraph() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 178 auto trans_op = AddTransOpNodeToGraph(func_graph, tuple_getitem, kernel_select, 0, false); in InsertTransOpForMultipleOutput() local
|