Home
last modified time | relevance | path

Searched defs:bias_size (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dbias_add_grad_impl.cu92 __global__ void BiasAddGradNCHW(const size_t size, const int batch, const int bias_size, const int … in BiasAddGradNCHW()
128 __global__ void FillDb(T *db, const size_t bias_size) { in FillDb()
135 void CalBiasAddGradNCHW(const size_t size, const size_t bias_size, const int height, const int widt… in CalBiasAddGradNCHW()
153 void CalBiasAddGradNHWC(const size_t size, const size_t bias_size, in CalBiasAddGradNHWC()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dconvolution_depthwise_fp32_coder.cc46 auto bias_size = static_cast<size_t>(channel * sizeof(float)); in InitWeightBias() local
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dconv_init_int8_wrapper.c43 size_t bias_size = up_round_oc * sizeof(int32_t); in ConvInit() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_depthwise_slidewindow_x86_fp32.cc190 auto bias_size = oc_algin * oc_tile_; in MallocWeightBiasData() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dconvolution_int8.cc72 size_t bias_size = up_round_oc * sizeof(int32_t); in InitWeightBias() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_fp16.cc83 auto bias_size = UP_ROUND(output_channel, C8NUM) * sizeof(float16_t); in MallocWeightBiasData() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dlstm_adjust_pass.cc157 size_t bias_size = kGateNums * hidden_size * kBiasNums * byte_num; in ReplaceLstmNode() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Ddepthwise_conv2d.cc197 size_t bias_size = C4NUM * CO4 * dtype_size; in InitBias() local
/third_party/ffmpeg/libavfilter/
Dvf_nnedi.c767 int bias_size = nns; in allocate_model() local