| /third_party/mindspore/tests/perf_test/ |
| D | test_lenet.py | 36 num_class = 10 variable
55 net = LeNet(num_class=num_class)
67 net = LeNet5(num_class=num_class)
72 sens = Tensor(np.ones([batch_size, num_class]).astype(np.float32))
|
| D | lenet.py | 53 def __init__(self, num_class=10, channel=1): argument
55 self.num_class = num_class
60 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/tests/st/networks/models/deeplabv3/src/ |
| D | miou_precision.py | 34 def __init__(self, num_class=21): argument
36 if not isinstance(num_class, int):
37 raise TypeError('num_class should be integer type, but got {}'.format(type(num_class)))
38 if num_class < 1:
39 raise ValueError('num_class must be at least 1, but got {}'.format(num_class))
40 self._num_class = num_class
|
| /third_party/mindspore/tests/st/quantization/lenet_quant/ |
| D | lenet_fusion.py | 33 def __init__(self, num_class=10, channel=1): argument
36 self.num_class = num_class
44 self.fc3 = nn.DenseBnAct(84, self.num_class)
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | model.py | 47 def __init__(self, num_class=10, channel=3): argument
49 self.num_class = num_class
54 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | ctcloss_cpu_kernel.cc | 41 size_t num_class, size_t batch_size, size_t b) { in InnerSoftMax() argument
46 for (size_t c = 0; c < num_class; ++c) { in InnerSoftMax()
47 if (inputs_addr[t * batch_size * num_class + b * num_class + c] > maxCoeff) { in InnerSoftMax()
48 maxCoeff = inputs_addr[t * batch_size * num_class + b * num_class + c]; in InnerSoftMax()
52 for (size_t c = 0; c < num_class; ++c) { in InnerSoftMax()
53 …sumCoeff += static_cast<T>(exp(inputs_addr[t * batch_size * num_class + b * num_class + c] - maxCo… in InnerSoftMax()
55 static_cast<T>(exp(inputs_addr[t * batch_size * num_class + b * num_class + c] - maxCoeff)); in InnerSoftMax()
58 for (size_t c = 0; c < num_class; ++c) { in InnerSoftMax()
|
| /third_party/mindspore/tests/st/gnn/ |
| D | gat.py | 65 num_class, argument
75 self.num_class = Validator.check_positive_int(num_class)
106 self.num_class,
|
| D | test_gat_model.py | 29 num_class = 7
39 num_class,
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | model.py | 48 def __init__(self, num_class=10, channel=3): argument
50 self.num_class = num_class
55 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/tests/st/fl/cross_silo_lenet/src/ |
| D | model.py | 48 def __init__(self, num_class=10, channel=3): argument
50 self.num_class = num_class
55 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/tests/st/hcom/ |
| D | hcom_sparsetensor.py | 36 def __init__(self, size=1024, batch_size=32, image_size=(3, 224, 224), num_class=10, argument
44 self.num_class = num_class
98 target = np.random.randint(0, self.num_class, size=(self.rank_size, self.rank_batch_size))
105 target_onehot = np.zeros(shape=(self.rank_batch_size, self.num_class))
147 …ds = FakeData(size=8, batch_size=batch_size, num_class=8, image_size=(), use_parallel=use_parallel)
|
| /third_party/mindspore/tests/st/ps/multi_full_ps/ |
| D | test_multi_full_ps.py | 67 def __init__(self, num_class=10, channel=3): argument
69 self.num_class = num_class
74 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/tests/st/probability/transforms/ |
| D | test_transform_bnn_model.py | 63 def __init__(self, num_class=10): argument
65 self.num_class = num_class
70 self.fc3 = fc_with_initialize(84, self.num_class)
|
| D | test_transform_bnn_layer.py | 64 def __init__(self, num_class=10): argument
66 self.num_class = num_class
71 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/tests/st/probability/bnn_layers/ |
| D | test_bnn_layer.py | 63 def __init__(self, num_class=10): argument
65 self.num_class = num_class
70 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/tests/st/probability/toolbox/ |
| D | test_uncertainty.py | 51 def __init__(self, num_class=10, channel=1): argument
53 self.num_class = num_class
58 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/tests/st/ops/ascend/ |
| D | test_sparseSoftmaxCrossEntropyWithLogits.py | 37 num_class = logits_shape[1]
38 labels = np.random.randint(low=0, high=num_class - 1, size=labels_shape).astype(np.int32)
41 features_reshape = np.reshape(features, [-1, num_class])
|
| /third_party/mindspore/tests/ut/python/pynative_mode/ |
| D | test_hook.py | 96 def __init__(self, num_class=10): argument
98 self.num_class = num_class
106 self.fc3 = fc_with_initialize(84, self.num_class)
150 label = Tensor(np.ones([net.batch_size, net.num_class]).astype(np.float32))
|
| /third_party/mindspore/tests/st/ps/full_ps/ |
| D | test_full_ps_lenet.py | 61 def __init__(self, num_class=10, channel=1): argument
63 self.num_class = num_class
68 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_hook.py | 74 def __init__(self, hook_function, cell_hook_function, num_class=10): argument
76 self.num_class = num_class
83 self.fc3 = fc_with_initialize(84, self.num_class)
180 label = Tensor(np.ones([net.batch_size, net.num_class]).astype(np.float32))
|
| D | test_pynative_lenet.py | 66 def __init__(self, num_class=10): argument
68 self.num_class = num_class
74 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/tests/st/fusion/ |
| D | test_conv_bn1_fusion.py | 27 num_class = 10 variable
69 self.dense = nn.Dense(output_channel, num_class)
97 self.dense = nn.Dense(output_channel, num_class)
123 self.dense = nn.Dense(output_channel, num_class)
|
| /third_party/mindspore/tests/st/nontask_sink/ |
| D | test_lenet.py | 65 def __init__(self, num_class=10): argument
67 self.num_class = num_class
73 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/tests/st/pynative/ms_function/ |
| D | test_pynative_lenet_ms_function.py | 103 def __init__(self, num_class=10): argument
105 self.num_class = num_class
111 self.fc3 = fc_with_initialize(84, self.num_class)
|
| /third_party/mindspore/tests/st/profiler/ |
| D | test_profiler.py | 59 def __init__(self, num_class=10, channel=1): argument
61 self.num_class = num_class
66 self.fc3 = fc_with_initialize(84, self.num_class)
|