Searched defs:weight_node (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | conv1d_weight_expanding_pass.cc | 27 lite::STATUS Conv1DWeightExpandingPass::ExpandFilterShape(const AnfNodePtr &weight_node, const sche… in ExpandFilterShape()
78 auto weight_node = conv_cnode->input(kConvWeightIndex); in Run() local
|
| D | group_depthwise_op_convert_pass.cc | 91 auto weight_node = conv_cnode->input(kConvWeightIndex); in Run() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | parser_utils.cc | 156 auto weight_node = cnode->input(index); in GetRealConvWeightNode() local
197 auto weight_node = cnode->input(index); in UnifyConvWeightFormat() local
220 int UnifyVariableConvWeight(const FuncGraphPtr &graph, const AnfNodePtr &weight_node, schema::Forma… in UnifyVariableConvWeight()
286 int UnifyConstConvWeight(const FuncGraphPtr &graph, const AnfNodePtr &weight_node, schema::Format s… in UnifyConstConvWeight()
323 int HandleConstConvWeightShared(const FuncGraphPtr &graph, const AnfNodePtr &weight_node, schema::F… in HandleConstConvWeightShared()
|
| D | conv1d_inout_adjust.cc | 66 lite::STATUS Conv1DInOutAdjust::ExpandFilterShape(const AnfNodePtr &weight_node, const schema::Form… in ExpandFilterShape()
137 auto weight_node = conv_cnode->input(kConvWeightIndex); in Run() local
|
| D | lstm_adjust_pass.cc | 46 auto weight_node = cnode->input(weight_index); in GetRealLstmWeightNode() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_deconv_parser.cc | 52 auto weight_node = GetConstInputNode(tf_node_map, tf_op.input(1)); in Parse() local
|
| D | tf_conv_parser.cc | 52 auto weight_node = GetConstInputNode(tf_node_map, tf_op.input(1)); in Parse() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/ |
| D | get_model_kernel.h | 119 auto weight_node = input_node->cast<ParameterPtr>(); in Init() local
|
| D | update_model_kernel.h | 102 auto weight_node = input_node->cast<ParameterPtr>(); in Init() local
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | fed_avg_kernel.h | 74 auto weight_node = in InitKernel() local
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | executor.cc | 314 …AnfNodePtr weight_node = AnfAlgo::VisitKernelWithReturnType(AnfAlgo::GetInputNode(cnode, weight_id… in GetTrainableParamName() local
|
| D | parameter_aggregator.cc | 374 …auto weight_node = AnfAlgo::VisitKernelWithReturnType(AnfAlgo::GetInputNode(cnode, cnode_weight_id… in JudgeRequiresAggr() local
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline.cc | 522 const CNodePtr &root_node, const AnfNodePtr &weight_node, in GetWeightInfo()
630 auto weight_node = cnode->input(weight_index); in FetchInfoForQuantExport() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | control_node_parser.cc | 40 for (const auto weight_node : weight_nodes) { in FetchWeightbyHostParameter() local
|