Searched refs:output_batch (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | pooling_grad.h | 26 …id MaxPoolingGrad(const float *input_ptr, const float *dy_ptr, float *output_ptr, int output_batch,
|
| D | pooling_grad.c | 103 …id MaxPoolingGrad(const float *input_ptr, const float *dy_ptr, float *output_ptr, int output_batch, in MaxPoolingGrad() argument
116 for (int ib = 0; ib < output_batch; ib++) { in MaxPoolingGrad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | pooling_grad.h | 26 …6Grad(const float16_t *input_ptr, const float16_t *dy_ptr, float16_t *output_ptr, int output_batch,
|
| D | pooling_grad.c | 105 …6Grad(const float16_t *input_ptr, const float16_t *dy_ptr, float16_t *output_ptr, int output_batch, in MaxPoolingFp16Grad() argument
119 for (int ib = 0; ib < output_batch; ib++) { in MaxPoolingFp16Grad()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | conv2d_3x3_int8_coder.cc | 85 int output_batch = conv_param_->output_batch_; in InitTmpBuffer() local
103 tmp_out_size_ = oc4 * C4NUM * output_batch * output_w * output_h * sizeof(uint8_t); in InitTmpBuffer()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | mirror_pad_impl.cuh | 41 …irrorPadGrad(const size_t dx_size, const size_t dy_size, T *dy, T *interim, const int output_batch,
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | convolution_3x3_int8.cc | 134 int output_batch = conv_param_->output_batch_; in InitTmpBuffer() local
169 size_t tmp_out_size = oc4 * C4NUM * output_batch * output_w * output_h * sizeof(uint8_t); in InitTmpBuffer()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pooling_int8.c | 33 int output_batch = pooling_param->output_batch_; in AvgPoolingInt8() local
43 for (int batch = 0; batch < output_batch; batch++) { in AvgPoolingInt8()
289 int output_batch = pooling_param->output_batch_; in MaxPoolingInt8() local
298 for (int batch = 0; batch < output_batch; batch++) { in MaxPoolingInt8()
|
| D | conv_depthwise_int8.c | 433 int8_t *output_batch = in ConvDw3x3Int8Pad() local
438 int8_t *output = output_batch; in ConvDw3x3Int8Pad()
456 output = output_batch + output_row_size; in ConvDw3x3Int8Pad()
469 …output = output_batch + output_row_size + (conv_param->output_w_ - 1) * conv_param->output_channel… in ConvDw3x3Int8Pad()
481 output = output_batch + (conv_param->output_h_ - 1) * output_row_size; in ConvDw3x3Int8Pad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pooling_fp32.c | 145 int output_batch = pooling_param->output_batch_; in MaxPooling() local
160 for (int batch = 0; batch < output_batch; batch++) { in MaxPooling()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pooling_fp16.c | 266 int output_batch = pooling_param->output_batch_; in MaxPoolingFp16() local
273 for (int batch = 0; batch < output_batch; batch++) { in MaxPoolingFp16()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | batch_op.cc | 400 TensorRow output_batch; in InvokeBatchMapFunc() local
412 output_batch.push_back(std::move(out)); in InvokeBatchMapFunc()
414 output->push_back(std::move(output_batch)); in InvokeBatchMapFunc()
|