| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | bias_add_grad_impl.cu | 92 __global__ void BiasAddGradNCHW(const size_t size, const int batch, const int bias_size, const int … in BiasAddGradNCHW() argument
105 int bg_offset = blockIdx.x % bias_size; in BiasAddGradNCHW()
106 int bg_id = blockIdx.x / bias_size; in BiasAddGradNCHW()
111 T val = *(dy + (img_id * bias_size + bg_offset) * img_size + img_offset); in BiasAddGradNCHW()
128 __global__ void FillDb(T *db, const size_t bias_size) { in FillDb() argument
129 …for (size_t pos = blockIdx.x * blockDim.x + threadIdx.x; pos < bias_size; pos += blockDim.x * grid… in FillDb()
135 void CalBiasAddGradNCHW(const size_t size, const size_t bias_size, const int height, const int widt… in CalBiasAddGradNCHW() argument
137 int batch_size = size / bias_size / height / width; in CalBiasAddGradNCHW()
141 int block_group_size = (block_num + bias_size - 1) / bias_size; in CalBiasAddGradNCHW()
142 block_num = block_group_size * bias_size; in CalBiasAddGradNCHW()
[all …]
|
| D | bias_add_grad_impl.cuh | 22 void CalBiasAddGradNHWC(const size_t size, const size_t bias_size,
25 void CalBiasAddGradNCHW(const size_t size, const size_t bias_size, const int height, const int widt…
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | convolution_depthwise_fp32_coder.cc | 46 auto bias_size = static_cast<size_t>(channel * sizeof(float)); in InitWeightBias() local
47 …bias_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, bias_size, kOfflinePackWe… in InitWeightBias()
49 MS_CHECK_RET_CODE(memset_s(bias_, bias_size, 0, bias_size), "memset bias failed!"); in InitWeightBias()
54 …MS_CHECK_RET_CODE(memcpy_s(bias_, bias_size, ori_bias, bias_tensor_->Size()), "memcpy_s bias faile… in InitWeightBias()
|
| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/ |
| D | conv_init_int8_wrapper.c | 43 size_t bias_size = up_round_oc * sizeof(int32_t); in ConvInit() local
62 bias_data_ = (int32_t *)(malloc(bias_size)); in ConvInit()
67 memset(bias_data_, 0, bias_size); in ConvInit()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_depthwise_slidewindow_x86_fp32.cc | 190 auto bias_size = oc_algin * oc_tile_; in MallocWeightBiasData() local
191 bias_data_ = malloc(bias_size * sizeof(float)); in MallocWeightBiasData()
196 memset(bias_data_, 0, bias_size * sizeof(float)); in MallocWeightBiasData()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | convolution_int8.cc | 72 size_t bias_size = up_round_oc * sizeof(int32_t); in InitWeightBias() local
95 bias_data_ = reinterpret_cast<int32_t *>(malloc(bias_size)); in InitWeightBias()
100 memset(bias_data_, 0, bias_size); in InitWeightBias()
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_lstm_op.py | 73 bias_size = num_directions * hidden_size * 4
74 weight_size = weight_size + bias_size
242 bias_size = 0 if not has_bias else num_directions * hidden_size * 4
247 bias_np = np.zeros([bias_size, 1, 1]).astype(np.float32)
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | depthwise_conv2d.cc | 197 size_t bias_size = C4NUM * CO4 * dtype_size; in InitBias() local
198 std::vector<char> temp_bias(bias_size, 0); in InitBias()
207 bias_data_ = allocator->Malloc(bias_size, temp_bias.data()); in InitBias()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_fp16.cc | 83 auto bias_size = UP_ROUND(output_channel, C8NUM) * sizeof(float16_t); in MallocWeightBiasData() local
84 bias_data_ = malloc(bias_size); in MallocWeightBiasData()
|
| /third_party/ffmpeg/libavfilter/ |
| D | vf_nnedi.c | 767 int bias_size = nns; in allocate_model() local
770 data = av_calloc(filter_size + bias_size, 4 * sizeof(float)); in allocate_model()
782 coeffs->softmax_bias_q1 = allocate(&data, bias_size); in allocate_model()
783 coeffs->elliott_bias_q1 = allocate(&data, bias_size); in allocate_model()
787 coeffs->softmax_bias_q2 = allocate(&data, bias_size); in allocate_model()
788 coeffs->elliott_bias_q2 = allocate(&data, bias_size); in allocate_model()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | lstm_adjust_pass.cc | 157 size_t bias_size = kGateNums * hidden_size * kBiasNums * byte_num; in ReplaceLstmNode() local
224 bias_size * num_directions, bias_shape, data_type, true, num_directions); in ReplaceLstmNode()
|