Searched defs:kernel_num (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_transform_fusion.cc | 55 int weight_shape_size, int kernel_num) { in GenerateNewWeightConv2D()
69 … const tensor::TensorPtr &weight_tensor, int64_t group, int kernel_num) { in GenerateNewWeightConv2DTranspose()
173 …Fusion::GenNewConvTensor(const FuncGraphPtr &func_graph, const CNodePtr &conv_node, int kernel_num, in GenNewConvTensor()
268 int kernel_num, const float *trans_scale) const { in CalNewWeightTensor()
313 int ConvTransformFusion::CalNewBiasTensor(float *bias_data, int kernel_num, bool bias_flag, const f… in CalNewBiasTensor()
|
| D | conv_bn_fusion.cc | 47 int kernel_num) { in CalTransale()
86 float *trans_bias, int kernel_num) { in CalTransBias()
158 int ConvBatchNormFusion::InitTransParam(const CNodePtr &bn_node, int kernel_num, float *trans_scale, in InitTransParam()
|
| D | conv_scale_fusion.cc | 42 int ConvScaleFusion::InitTransParam(const CNodePtr &scale_node, int kernel_num, float *trans_scale, in InitTransParam()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | gpu_session.cc | 550 int kernel_num = kernel_graph->execution_order().size(); in ExecuteGraph() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/common/ |
| D | gllo_utils.cc | 526 ParameterPtr AddNewBiasNode(float *bias_data, const FuncGraphPtr &func_graph, int kernel_num, TypeI… in AddNewBiasNode()
|