Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/random/
Drandom_categorical_gpu_kernel.h151 size_t num_classes_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Ddetection_post_process_parameter.h32 int64_t num_classes_; member
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Ddataset_op.h389 int64_t num_classes_; // Number of classes variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dtransforms.cc154 int32_t num_classes_; member