Home
last modified time | relevance | path

Searched defs:cast_cnode (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/import/
Dcast_op_adjust.cc25 bool GetInOutDataTypeValue(const CNodePtr &cast_cnode, int *output_type_value, int *input_type_valu… in GetInOutDataTypeValue()
140 auto cast_cnode = node->cast<CNodePtr>(); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Deliminate_redundant_cast_pass.cc24 auto cast_cnode = anf_node->cast<CNodePtr>(); in RemoveCastOp() local
Dkvcache_quant_pass.cc111 auto cast_cnode = anf_node->cast<CNodePtr>(); in ReplaceCastOpToQuantOp() local
136 auto cast_cnode = anf_node->cast<CNodePtr>(); in ReplaceCastOpToAntiQuantOp() local
Dint64_cast_int32_pass.cc124 auto cast_cnode = graph->NewCNode({value_node}); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dload_eliminate.cc50 auto cast_cnode = dyn_cast<CNode>(input_load); in operator ()() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Doptimize_parallel_allgather_comm.cc39 const CNodePtr &cast_cnode) { in MoveCastBehindAllGather()
Dsplit_layernorm_comm_fp.cc313 auto cast_cnode = manager->node_users()[tuple_get_item_cnode].front().first->cast<CNodePtr>(); in SplitIntoInterleaved() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dfused_cast_add.cc53 auto cast_cnode = cast_node->cast<CNodePtr>(); in SelectAddInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dadjust_depend_for_parallel_optimizer_recompute_all_gather.cc141 auto cast_cnode = depend_node->cast<CNodePtr>(); in AdjustAllgatherDepend() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dresize_fusion.cc227 auto cast_cnode = div_cnode->input(1)->cast<CNodePtr>(); in DoFuison() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dinsert_quant_node_manager.cc57 const CNodePtr &cast_cnode) { in SetCastNodeAbstract()
863 CNodePtr cast_cnode = nullptr; in InsertAscendAntiQuantNode() local
1094 auto cast_cnode = func_graph->NewCNode(cast_op_inputs); in NewCastNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Ddynamic_gru_v2_grad_fission.cc76 auto cast_cnode = NewCNode(cast_inputs, func_graph); in CastNodeToSpecificDType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc814 auto cast_cnode = cast_node->cast<CNodePtr>(); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc1199 auto cast_cnode = graph->NewCNode({value_node}); in GenCastNode() local