Home
last modified time | relevance | path

Searched defs:num_classes_ (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Drandom_categorical.h35 int num_classes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dfake_image_node.h93 int32_t num_classes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dfake_image_op.h98 int32_t num_classes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dnll_loss_gpu_kernel.h64 unsigned int num_classes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/random/
Drandom_categorical_gpu_kernel.h162 size_t num_classes_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Ddetection_post_process_parameter.h32 int64_t num_classes_; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Ddataset_op.h415 int64_t num_classes_; // Number of classes variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/
Dtransforms.cc152 int32_t num_classes_; member