Searched defs:bn_node (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/ |
| D | expand_dims_for_batchnorm.cc | 92 bool ExpandInputDims(const FuncGraphPtr &graph, const CNodePtr &bn_node, size_t index, in ExpandInputDims()
117 bool ExpandInputDims(const FuncGraphPtr &graph, const CNodePtr &bn_node, const ExpandDimsInfo &expa… in ExpandInputDims()
130 void ExpandSingleOutputDims(const FuncGraphPtr &graph, const CNodePtr &bn_node, const FuncGraphMana… in ExpandSingleOutputDims()
139 void ExpandMultiOutputDims(const FuncGraphPtr &graph, const CNodePtr &bn_node, const FuncGraphManag… in ExpandMultiOutputDims()
166 void ExpandOutputDims(const FuncGraphPtr &graph, const CNodePtr &bn_node, const ExpandDimsInfo &exp… in ExpandOutputDims()
195 CNodePtr bn_node = node->cast<CNodePtr>(); in Process() local
|
| D | batchnorm_transform.cc | 135 void SetScopeForNewNodes(const vector<AnfNodePtr> &nodes, const AnfNodePtr &bn_node) { in SetScopeForNewNodes()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/ |
| D | conv_bn_fusion_test.cc | 86 auto bn_node = std::make_unique<schema::CNodeT>(); in BuildCaffeGraph() local
167 auto bn_node = std::make_unique<schema::CNodeT>(); in BuildTFGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_bn_fusion.cc | 50 int ConvBatchNormFusion::InitTransParam(const CNodePtr &bn_node, int kernel_num, float *trans_scale, in InitTransParam()
|
| D | squeeze_fusion.cc | 71 auto bn_node = act_cnode->input(1); in Process() local
|
| D | batchnorm_to_scale_fusion.cc | 155 int CalculateScaleAndBiasFromBN(const CNodePtr &bn_node, int kernel_num, float *trans_scale, float … in CalculateScaleAndBiasFromBN()
|