Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16_grad/
Dconvolution_grad_filter.c30 int k_h = conv_param->kernel_h_; in FilterGrad32Arm() local
108 int k_h = conv_param->kernel_h_; in FilterGrad16Arm() local
166 int k_h = conv_param->kernel_h_; in FilterGrad8Arm() local
215 int k_h = conv_param->kernel_h_; in FilterGrad4Arm() local
263 int k_h = conv_param->kernel_h_; in FilterGradLeftoverArm() local
313 int k_h = conv_param->kernel_h_; in ConvDwFilterFp16Grad() local
Dconvolution_grad_input.c32 int k_h = conv_param->kernel_h_; in ConvDwInputGrad16() local
137 int k_h = conv_param->kernel_h_; in ConvDwInputGrad8() local
220 int k_h = conv_param->kernel_h_; in ConvDwInputGrad4() local
296 int k_h = conv_param->kernel_h_; in ConvDwInputGradFp16() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dconvolution_grad_filter.c28 int k_h = conv_param->kernel_h_; in FilterGrad16Arm() local
105 int k_h = conv_param->kernel_h_; in FilterGrad12Arm() local
175 int k_h = conv_param->kernel_h_; in FilterGrad8Arm() local
234 int k_h = conv_param->kernel_h_; in FilterGrad4Arm() local
284 int k_h = conv_param->kernel_h_; in Filtergrad2Arm() local
332 int k_h = conv_param->kernel_h_; in ConvDwFilterGrad() local
Dconvolution_grad_input.c31 int k_h = conv_param->kernel_h_; in ConvDwInputGrad() local
/third_party/mesa3d/src/gallium/targets/teflon/
Dtest_model_generation.py41 def create_model_keras(batch_size, in_w, in_h, k_w, k_h, in_ch, out_ch, stride, padding, signed, se… argument
138 def create_model(batch_size, in_w, in_h, k_w, k_h, in_ch, out_ch, stride, padding, signed, seed, de… argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/
Dsplit_strategy.cc37 int64_t k_h = weight_shape.at(kShapeH); in ApproximateFLOPs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/
Ddeconvolution_grad_filter.cc92 int k_h = conv_param->kernel_h_; in DoExecute() local
Dconvolution_grad_input.cc96 int k_h = conv_param->kernel_h_; in DoExecute() local
Dconvolution.cc89 const int k_h = conv_param_->kernel_h_; in DoExecute() local
Dconvolution_grad_filter.cc98 int k_h = conv_param->kernel_h_; in DoExecute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Ddeformable_offsets_fusion.cc63 for (int64_t k_h = 0; k_h < K_H; ++k_h) { in CreateHelperNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16_grad/
Dconvolution_fp16_grad_input.cc103 int k_h = conv_param->kernel_h_; in DoExecute() local
Dconvolution_fp16_grad_filter.cc102 int k_h = conv_param->kernel_h_; in DoExecute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dop_node.cc642 auto k_h = (kernel_size[0] - 1) * dilation[2] + 1; in InferShape() local
659 auto k_h = weight_shape[2]; in InferShape() local