Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/string/
Dextract_feature.cc57 auto label_data = reinterpret_cast<int32_t *>(out_tensors_.at(0)->MutableData()); in Run() local
58 CHECK_NULL_RETURN(label_data); in Run()
67 label_data[i] = 0; in Run()
72 label_data[i] = hash_value; in Run()
76 label_data[0] = 0; in Run()
/third_party/mindspore/tests/st/heterogeneous_excutor/
Dtest_heterogeneous_excutor.py67 label_data = np.random.randn(32, 10).astype(np.float32)
76 out = train_network(Tensor(input_data), Tensor(label_data))
89 out_heter = train_network_heter(Tensor(input_data), Tensor(label_data))
/third_party/mindspore/tests/st/mix_precision/
Dtest_mix_precision.py83 label_data = np.random.randn(32, 10).astype(np.float32)
92 out = train_network(Tensor(input_data), Tensor(label_data))
104 out_pynative = train_network_pynative(Tensor(input_data), Tensor(label_data))
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dvoc_op.cc286 std::vector<uint32_t> label_data, difficult_data, truncate_data; in ReadAnnotationToTensor() local
291 label_data.push_back(static_cast<uint32_t>(class_index_[item.first])); in ReadAnnotationToTensor()
293 label_data.push_back(static_cast<uint32_t>(label_index_[item.first])); in ReadAnnotationToTensor()
307 RETURN_IF_NOT_OK(Tensor::CreateFromVector(label_data, TensorShape({bbox_num, 1}), &label)); in ReadAnnotationToTensor()
/third_party/mindspore/tests/st/auto_parallel/
Donehot_model_parallel.py93 def label_data(self, shape, classes): member in DataGenerator
105 self.label_full, self.label_part = data_gen.label_data((batch_size,), classes)
Dsoft_entropy_loss_expand_parallel.py87 def label_data(self, shape, embed_): member in DataGenerator
191 self.label_full, self.label_part = data_gen.label_data((batch_size,), embed)
Dresnet50_expand_loss.py313 def label_data(self, shape): member in DataGenerator
367 _, label_part = data_gen.label_data((32 * 8,))
/third_party/mindspore/tests/ut/python/mindrecord/
Dutils.py83 label_data = file_reader.read()
87 "label_data": label_data,