| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_ops_one_hot_ext.py | 24 def generate_random_input(num_classes): argument
29 def onehot_forward_func(tensor, num_classes): argument
37 def call_onehot(tensor, num_classes): argument
46 def onehot_backward_func(tensor, num_classes): argument
53 def construct(self, x, num_classes): argument
281 def _foreach_run(inputs, num_classes, batch): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | random_categorical.cu | 22 … const size_t batch_size, const size_t num_classes, S *output_addr) { in RandomCategorical()
38 …GetCdf(const T *logits_addr, double **dev_cdf, const size_t batch_size, const size_t num_classes) { in GetCdf()
62 … const size_t batch_size, const size_t num_classes, S *output_addr, in RandomCategoricalKernel()
71 …dfKernel(const T *logits_addr, double **dev_cdf, const size_t batch_size, const size_t num_classes, in GetCdfKernel()
|
| D | combined_non_max_suppression_impl.cu | 81 __global__ void presort(int num_classes, int num_boxes, int batch_size, T *scores, float *new_score… in presort()
96 __global__ void Init(int num_classes, int num_boxes, int batch_size, bool *mask) { in Init()
104 __global__ void nms(int batch_size, int num_classes, T *iou_threshold, bool *sel, float *boxes_resu… in nms()
128 __global__ void nmsReducePass(int batch_size, int num_classes, bool *sel, int *index, int num_boxes… in nmsReducePass()
144 __global__ void sizeperclass(int batch_size, int num_classes, bool *sel, int num_boxes, int *index, in sizeperclass()
164 int num_classes, int num_boxes, int q) { in output()
201 cudaError_t CalSort(T *scores, int *index, T *score_threshold, int num_classes, T *boxes, float *ne… in CalSort()
220 cudaError_t Calnms(int batch_size, int num_classes, T *iou_threshold, bool *sel, float *boxes_resul… in Calnms()
245 … int num_classes, int num_boxes, int q, const uint32_t &device_id, cudaStream_t cuda_stream) { in Caloutput()
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/ |
| D | test_lenet.py | 226 def test_train_lenet_with_new_interface(num_classes=10, epoch=20, batch_size=32): argument
253 def test_train_lenet_with_new_interface_tuple(num_classes=10, epoch=20, batch_size=32): argument
285 def test_train_lenet_with_new_interface_list(num_classes=10, epoch=20, batch_size=32): argument
|
| D | test_gpu_resnet.py | 287 def __init__(self, block, layer_num, num_classes=100): argument
326 def resnet50(num_classes): argument
333 def test_trainTensor(num_classes=10, epoch=8, batch_size=1): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/fasterrcnn/src/ |
| D | label_assignment.py | 101 num_classes=80, argument
231 …posal_target(rois, rois_mask, gts, rois_per_batch, fg_fraction, fg_thresh, bg_thresh, num_classes): argument
297 num_classes, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | random_categorical_cpu_kernel.cc | 83 void GetCdf(T *logits_addr, T *dev_cdf, const size_t batch_size, const size_t num_classes) { in GetCdf()
112 const size_t num_classes, T2 *output_addr) { in RandomCategoricalFunc()
151 int num_classes = input_shape_[input_shape_.size() - 1]; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | one_hot_op.h | 30 …OneHotOp(int num_classes, double smoothing_rate) : num_classes_(num_classes), smoothing_rate_(smoo… in OneHotOp()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | omniglot_op.cc | 91 …OmniglotOp::CountRowsAndClasses(const std::string &path, int64_t *num_rows, int64_t *num_classes) { in CountRowsAndClasses()
119 Status OmniglotOp::GetNumClasses(int64_t *num_classes) { in GetNumClasses()
|
| D | lsun_op.cc | 104 … const std::vector<std::string> &classes, int64_t *num_rows, int64_t *num_classes) { in CountRowsAndClasses()
157 Status LSUNOp::GetNumClasses(int64_t *num_classes) { in GetNumClasses()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_batchnorm_batch_parallel.py | 95 def __init__(self, num_classes=100): argument
117 def batchnorm_net(num_classes): argument
|
| D | test_operator_model_parallel.py | 295 def __init__(self, block, num_classes=100): argument
320 def __init__(self, block, num_classes=100): argument
340 def resnet_operator_net(num_classes): argument
344 def resnet_model_parallel_net(num_classes): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/parser/caffe/ |
| D | caffe_decbbox_parser.cc | 47 uint32_t num_classes = proto.num_classes(); in Parse() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/ |
| D | alexnet.py | 20 def __init__(self, num_classes=10): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/examples/export_models/models/ |
| D | NetworkInNetwork.py | 26 def __init__(self, num_classes=10, num_channel=3): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_sampled_softmax_loss_op.py | 23 def generate_test_data(num_classes, batch_size, sampled): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/deeplabv3/src/ |
| D | deeplabv3.py | 253 num_classes, argument
348 …def __init__(self, num_classes, feature_shape, backbone, channel, depth, infer_scale_sizes, atrous… argument
426 def deeplabv3_resnet50(num_classes, feature_shape, image_pyramid, argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/resnet50/src/ |
| D | CrossEntropySmooth.py | 26 def __init__(self, sparse=True, reduction='mean', smooth_factor=0., num_classes=1000): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/model/ |
| D | res18_example.py | 103 def __init__(self, block, num_classes=100): argument
172 def __init__(self, block, num_classes=100): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/extend/ |
| D | array_func.py | 177 def one_hot(tensor, num_classes): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/mindrecord/meta/ |
| D | shard_category.cc | 39 int64_t ShardCategory::GetNumSamples(int64_t dataset_size, int64_t num_classes) { in GetNumSamples()
|
| D | shard_sequential_sample.cc | 27 int64_t ShardSequentialSample::GetNumSamples(int64_t dataset_size, int64_t num_classes) { in GetNumSamples()
|
| /third_party/mindspore/mindspore-src/source/tests/st/pynative/network/ |
| D | test_pynative_resnet50_ascend.py | 294 def __init__(self, block, layer_num, num_classes=100): argument
333 def resnet50(num_classes): argument
342 def test_train_tensor(num_classes=10, epoch=3, batch_size=1): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/infer/ |
| D | dpico_decbbox_infer.cc | 61 uint32_t num_classes = 0; in Infer() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/dataset/ |
| D | c_api_dataset_omniglot_test.cc | 204 int64_t num_classes = ds->GetNumClasses(); in TEST_F() local
244 int64_t num_classes = ds->GetNumClasses(); in TEST_F() local
|