| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | batchnorm_fp32_coder.cc | 34 int n_dim = static_cast<int>(input_shapes.size()); in Init() local
|
| D | softmax_fp32_coder.cc | 30 int n_dim = softmax_param_->n_dim_; in Prepare() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | softmax_grad.c | 24 int n_dim = parameter->n_dim_; in SoftmaxGrad() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | prelu_fp32.cc | 76 auto n_dim = in_shape.size(); in ReSize() local
|
| D | softmax_fp32.cc | 51 auto n_dim = softmax_param_->n_dim_; in ReSize() local
|
| D | log_softmax_fp32.cc | 56 auto n_dim = softmax_param_->n_dim_; in ReSize() local
|
| D | batchnorm_fp32.cc | 65 auto n_dim = input_shapes.size(); in FillParam() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | softmax_fp16.cc | 51 auto n_dim = softmax_param_->n_dim_; in ReSize() local
|
| D | log_softmax_fp16.cc | 58 auto n_dim = softmax_param_->n_dim_; in ReSize() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | softmax_int8.c | 23 int n_dim = parameter->n_dim_; in SoftmaxInt8() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | stack_base.cc | 36 static int GetCopyNum(const std::vector<int> &in_shape, int axis, int n_dim) { in GetCopyNum()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | log_softmax_fp32.c | 48 int n_dim = parameter->n_dim_; in LogSoftmax() local
|
| D | softmax_fp32.c | 107 int n_dim = parameter->n_dim_; in Softmax() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | log_softmax_fp16.c | 51 int n_dim = parameter->n_dim_; in LogSoftmaxFp16() local
|
| D | softmax_fp16.c | 97 int n_dim = parameter->n_dim_; in SoftmaxFp16() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | sparse_softmax_cross_entropy_with_logits.cc | 130 int n_dim = sm_params_.n_dim_; in Run() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | batchnorm_int8_coder.cc | 29 size_t n_dim = input_shapes.size(); in Prepare() local
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/operator_library/kernels/nnacl/fp32/ |
| D | softmax_fp32.c | 107 int n_dim = parameter->n_dim_; in Softmax() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | batchnorm_int8.cc | 173 auto n_dim = input_shapes.size(); in Init() local
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.cc | 1979 auto n_dim = args.host_shape[kN]; in NchwFracZTransWithGroups() local
|