Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_3x3_int8_coder.cc85 int output_batch = conv_param_->output_batch_; in InitTmpBuffer() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dconvolution_3x3_int8.cc134 int output_batch = conv_param_->output_batch_; in InitTmpBuffer() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpooling_grad.c105 …6Grad(const float16_t *input_ptr, const float16_t *dy_ptr, float16_t *output_ptr, int output_batch, in MaxPoolingFp16Grad()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpooling_grad.c103 …id MaxPoolingGrad(const float *input_ptr, const float *dy_ptr, float *output_ptr, int output_batch, in MaxPoolingGrad()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpooling_int8.c33 int output_batch = pooling_param->output_batch_; in AvgPoolingInt8() local
289 int output_batch = pooling_param->output_batch_; in MaxPoolingInt8() local
Dconv_depthwise_int8.c433 int8_t *output_batch = in ConvDw3x3Int8Pad() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpooling_fp32.c145 int output_batch = pooling_param->output_batch_; in MaxPooling() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dbatch_op.cc400 TensorRow output_batch; in InvokeBatchMapFunc() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpooling_fp16.c266 int output_batch = pooling_param->output_batch_; in MaxPoolingFp16() local