Searched defs:kernel_num (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_transform_fusion.cc | 65 size_t weight_shape_size, int kernel_num) { in GenerateNewWeightConv2D()
79 … const tensor::TensorPtr &weight_tensor, int64_t group, int kernel_num) { in GenerateNewWeightConv2DTranspose()
100 … const tensor::TensorPtr &weight_tensor, int64_t group, int kernel_num) { in GenerateNewWeightConv2DTranspose_NCHW()
228 …Fusion::GenNewConvTensor(const FuncGraphPtr &func_graph, const CNodePtr &conv_node, int kernel_num, in GenNewConvTensor()
323 int kernel_num, const float *trans_scale) const { in CalNewWeightTensor()
374 int ConvTransformFusion::CalNewBiasTensor(float *bias_data, int kernel_num, bool bias_flag, const f… in CalNewBiasTensor()
|
| D | batchnorm_to_scale_fusion.cc | 52 int kernel_num) { in CalTransale()
91 float *trans_bias, int kernel_num) { in CalTransBias()
155 int CalculateScaleAndBiasFromBN(const CNodePtr &bn_node, int kernel_num, float *trans_scale, float … in CalculateScaleAndBiasFromBN()
|
| D | conv_bn_fusion.cc | 50 int ConvBatchNormFusion::InitTransParam(const CNodePtr &bn_node, int kernel_num, float *trans_scale, in InitTransParam()
|
| D | conv_scale_fusion.cc | 44 int ConvScaleFusion::InitTransParam(const CNodePtr &scale_node, int kernel_num, float *trans_scale, in InitTransParam()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | super_kernel_actor.cc | 319 size_t kernel_num = kernel_actors_.size(); in UpdateMemoryTraceMangerStatus() local
442 size_t kernel_num = kernel_actors_.size(); in RunGraphKernelByKernel() local
801 size_t kernel_num = execution_order.size(); in BuildKernelActors() local
896 size_t kernel_num = execution_order.size(); in ParseInputIndex() local
936 size_t kernel_num = execution_order.size(); in CalcRefCount() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | fused_batch_norm.c | 50 … const void *mean_data, const void *var_data, float eps, int kernel_num) { in FusedBatchNormCalculateScaleF32()
68 … const void *mean_data, const void *var_data, float eps, int kernel_num) { in FusedBatchNormCalculateScaleF16()
105 … const void *mean_data, const void *var_data, float eps, int kernel_num) { in FusedBatchNormBatchnorm2Scale()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/session/ |
| D | dynamic_session.cc | 149 int kernel_num = static_cast<int>(kernels_.size()); in RunKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | pack_weight_manager.cc | 157 auto kernel_num = model->graph_.all_nodes_.size(); in StoreOriginTensorData() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/ |
| D | gpu_session.cc | 537 int kernel_num = kernel_graph->execution_order().size(); in ExecuteGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/ |
| D | gllo_utils.cc | 487 ParameterPtr AddNewBiasNode(const float *bias_data, const FuncGraphPtr &func_graph, int kernel_num,… in AddNewBiasNode()
|