Home
last modified time | relevance | path

Searched refs:label_np (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/tests/ut/python/exec/
Dtest_train.py32 def me_train_tensor(net, input_np, label_np, epoch_size=2): argument
41 label_np = np.argmax(label_np, axis=-1).astype(np.int32)
44 _train_net(Tensor(input_np), Tensor(label_np))
79 label_np = np.ones([1, 3, 3, 3], np.int32) * 2
80 me_train_tensor(Net(3, bias_init=bias_init), input_np, label_np)
111 label_np = np.ones([1, 6, 32, 32]).astype(np.int32)
112 me_train_tensor(net, input_np, label_np)
148 label_np = np.ones([32, 12]).astype(np.int32)
149 me_train_tensor(net, input_np, label_np)
178 label_np = np.ones([32, 16]).astype(np.int32)
[all …]
Dtest_train_with_lars.py79 def me_train_tensor(net, input_np, label_np, epoch_size=2): argument
89 label = Tensor(label_np)
/third_party/mindspore/tests/st/fusion/
Dtest_conv_bn1_fusion.py41 def me_train_tensor(net, input_np, label_np, epoch_size=2): argument
52 output = _train_net(Tensor(input_np), Tensor(label_np))
83 label_np = np.ones([batch_size]).astype(np.int32)
84 me_train_tensor(net, input_np, label_np)
110 label_np = np.ones([batch_size]).astype(np.int32)
111 me_train_tensor(net, input_np, label_np)
135 label_np = np.ones([batch_size]).astype(np.int32)
136 me_train_tensor(net, input_np, label_np)
/third_party/mindspore/tests/ut/python/parallel/
Dtest_auto_parallel_resnet.py296 label_np = np.zeros([batch_size]).astype(np.int32)
298 label_np[i] = i % num_classes
299 dataset = DatasetLenet(Tensor(input_np), Tensor(label_np), 1)
672 label_np = np.zeros([batch_size]).astype(np.int32)
674 label_np[i] = i % num_classes
675 dataset = DatasetLenet(Tensor(input_np), Tensor(label_np), 1)
700 label_np = np.zeros([batch_size]).astype(np.int32)
702 label_np[i] = i % num_classes
703 dataset = DatasetLenet(Tensor(input_np), Tensor(label_np), 1)
726 label_np = np.zeros([batch_size]).astype(np.int32)
[all …]
Dtest_bias_add.py88 label_np = np.zeros([16, 2048]).astype(np.float32)
89 dataset = DatasetLenet(Tensor(input_np), Tensor(label_np), 1)
Dtest_auto_parallel_resnet_sharding_propagation2.py297 label_np = np.zeros([batch_size]).astype(np.int32)
299 label_np[i] = i % num_classes
300 dataset = DatasetLenet(Tensor(input_np), Tensor(label_np), 1)
Dtest_auto_parallel_resnet_sharding_propagation.py299 label_np = np.zeros([batch_size]).astype(np.int32)
301 label_np[i] = i % num_classes
302 dataset = DatasetLenet(Tensor(input_np), Tensor(label_np), 1)
/third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/
Dtest_apply_adam.py41 label_np = np.random.randint(self.output_channels, size=self.batch_num)
43 label_np_onehot[np.arange(self.batch_num), label_np] = 1.0