Searched refs:input_n_ (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | conv3d_transpose_gpu_kernel.h | 55 …CalPad3d(input_padded_size_ / sizeof(T), input_addr, input_n_, input_c_, input_old_depth_, input_o… in Launch()
82 …CalPad3d(output_size_ / sizeof(T), stride_padded, input_n_, input_c_, stride_pad_depth_, stride_pa… in Launch()
137 …SetNCDHW(input_shape, &input_n_, &input_c_, &input_old_depth_, &input_old_height_, &input_old_widt… in Init()
251 input_n_ = 0; in ResetResource()
494 IntToSize(input_n_), IntToSize(input_c_), IntToSize(input_old_depth_ + padding_diff[0]), in UsePadProcess()
558 int input_n_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | roi_pooling_fp32.h | 34 int input_n_; member
|
| D | roi_pooling_fp32.c | 36 int batch_size = param->input_n_; in ROIPooling()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | roi_pooling_fp32.cc | 60 param_->input_n_ = in_shape.at(0); in ReSize()
|