Home
last modified time | relevance | path

Searched defs:batch_ (Results 1 – 25 of 27) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/dynamic_parameter/
Dmatmul_dynamic_parameter.h23 std::string batch_; member
Ddynamic_lstm_parameter.h23 std::string batch_; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dnllloss.h26 int batch_; member
Dcrop_and_resize.h27 int batch_; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcdist_grad_cpu_kernel.h49 int64_t batch_; variable
Dcdist_cpu_kernel.h52 int64_t batch_; variable
Dmax_pool_grad_with_argmax_cpu_kernel.h57 int batch_ = 0; variable
Dmax_pool_with_argmax_cpu_kernel.h58 int batch_ = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dmaxpool_grad_grad_with_argmax_gpu_kernel.h55 int batch_ = 0; variable
Dbatch_norm_grad_gpu_kernel.h94 int batch_; variable
Dmaxpool_grad_grad_gpu_kernel.h67 int batch_ = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/online_fusion/
Dreduce_concat_fp32.h41 int64_t batch_ = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dinstance_norm_parameter.h27 int batch_; member
Dgru_parameter.h28 int batch_; member
Dgroup_norm_parameter.h28 int batch_; member
Dattention_parameter.h39 int batch_; // batch of query/key/value/position member
Dlstm_parameter.h30 int batch_; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dattention_fp32.h32 int batch_; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Dmatmul_gpu_kernel.h86 size_t batch_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dlstm_grad_fp32.h29 int batch_; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/quant/
Dbatchnorm_fold_grad_gpu_kernel.h124 int batch_; variable
Dbatchnorm_fold_gpu_kernel.h179 int batch_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Dresizebicubic_helper.h162 int batch_; variable
Dresizebicubicgrad_helper.h174 int batch_; variable
Dscale_and_translate_grad_helper.h271 int64_t batch_; variable

12