Home
last modified time | relevance | path

Searched defs:output_hw (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dconv_fp16.c33 int output_hw = conv_param->output_h_ * conv_param->output_w_; in ConvFp16() local
75 int output_hw = conv_param->output_h_ * conv_param->output_w_; in ConvOutNc8hw8Fp16() local
/third_party/mindspore/mindspore/core/ops/
Dconv2d.cc72 void Conv2DPadFunction(std::vector<int64_t> *output_hw, std::vector<int64_t> *pad_list, const int64… in Conv2DPadFunction()
208 std::vector<int64_t> output_hw; in Conv2dInferShape() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_nn.cc203 void Conv2DPadFunction(std::vector<int64_t> *output_hw, std::vector<int64_t> *pad_list, const int64… in Conv2DPadFunction()
300 std::vector<int64_t> output_hw; in InferImplConv2D() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_common_fp32.c34 int output_hw = conv_param->output_h_ * conv_param->output_w_; in ConvFp32() local
116 int output_hw = conv_param->output_h_ * conv_param->output_w_; in ConvFp32OutNC4HW4() local