Home
last modified time | relevance | path

Searched refs:dataset_types (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore/train/
Ddataset_helper.py63 …def __init__(self, network, dataset_types, dataset_shapes, queue_name, min_shapes=None, max_shapes… argument
67 self.info = (dataset_types, dataset_shapes)
69 self.get_next = P.GetNext(dataset_types, dataset_shapes, len(dataset_types), queue_name)
82 def _generate_dataset_sink_mode_net(network, dataset_shapes, dataset_types, queue_name, argument
85 … network = _DataWrapper(network, dataset_types, dataset_shapes, queue_name, min_shapes, max_shapes)
97 dataset_types, dataset_shapes = dataset_helper.types_shapes()
100 network = _generate_dataset_sink_mode_net(network, dataset_shapes, dataset_types,
155 dataset_types, dataset_shapes = dataset_helper.get_data_info()
156 dataset_types = [pytype_to_dtype(x) for x in dataset_types]
158 key = str(dataset_types) + str(dataset_shapes)
[all …]
D_utils.py54 dataset_types = _convert_type(dataset.output_types())
56 return dataset_types, dataset_shapes
65 dataset_types, dataset_shapes = _get_types_and_shapes(exec_dataset)
72 dataset_types,
166 def _construct_input_tensors(dataset_types, dataset_shapes, device_number=1): argument
168 tensor_list_run = _construct_tensor_list(dataset_types, dataset_shapes, batch_expand_num=1)
169 …tensor_list_compile = _construct_tensor_list(dataset_types, dataset_shapes, batch_expand_num=devic…
/third_party/mindspore/tests/ut/python/train/
Dtest_amp.py97 def __init__(self, dataset_types, dataset_shapes): argument
99 np_types=dataset_types,
114 dataset_types = (np.float32, np.float32)
117 dataset = MindDataSet(dataset_types, dataset_shapes)
131 dataset_types = (np.float32, np.float32)
134 dataset = MindDataSet(dataset_types, dataset_shapes)
148 dataset_types = (np.float32, np.float32)
155 dataset = MindDataSet(dataset_types, dataset_shapes)
Dtest_training.py82 dataset_types = (np.float32, np.float32)
86 np_types=dataset_types,
136 dataset_types = (np.float32, np.float32)
140 np_types=dataset_types,
166 dataset_types = (np.float32, np.float32)
170 np_types=dataset_types,
255 dataset_types = (np.float32, np.float32)
259 np_types=dataset_types,
Dtest_dataset_helper.py26 dataset_types = (np.int32, np.int32, np.int32, np.int32, np.int32, np.int32, np.int32)
30 dataset = MindData(size=2, batch_size=batch_size, np_types=dataset_types,
/third_party/mindspore/tests/st/data_transfer/
Dtest_tdt_data_transfer.py90 dataset_types = convert_type(dataset_shapes, np_types)
91 return dataset_shapes, dataset_types
105 def __init__(self, network, dataset_types, dataset_shapes, shared_name=''): argument
107 self.get_next = P.GetNext(dataset_types, dataset_shapes, len(dataset_shapes), shared_name)
116 dataset_shapes, dataset_types = get_dataset_shapes_and_types(dataset)
122 net_with_dataset = NetWithTDT(net, dataset_types, dataset_shapes, queue_name)
124 …_cell_graph_executor.init_dataset(dataset.queue_name, 1, batch_size, dataset_types, dataset_shapes…
/third_party/mindspore/tests/st/networks/models/resnet50/src_thor/
Ddataset_helper.py88 dataset_types, dataset_shapes = _get_types_and_shapes(dataset)
89 self.dataset_types, self.dataset_shapes = dataset_types, dataset_shapes
102 return self.dataset_types, self.dataset_shapes
Dmodel_thor.py58 dataset_types = _convert_type(dataset.output_types())
60 return dataset_types, dataset_shapes
69 dataset_types, dataset_shapes = _get_types_and_shapes(exec_dataset)
73 dataset_types,
/third_party/mindspore/tests/ut/python/parallel/
Dtest_auto_parallel_flag.py32 def __init__(self, dataset_types, dataset_shapes): argument
34 np_types=dataset_types,
99 dataset_types = (np.float32, np.float32)
102 dataset = MindDataSet(dataset_types, dataset_shapes)
/third_party/mindspore/tests/st/ops/ascend/
Dtest_tdt_data_ms.py78 dataset_types = convert_type(dataset_shapes, np_types) variable
81 get_next = P.GetNext(dataset_types, dataset_shapes, 2, ds1.queue_name)
96 dataset_types, dataset_shapes, (), 'dataset')
/third_party/mindspore/mindspore/train/train_thor/
Dmodel_thor.py52 dataset_types = _convert_to_ms_type(dataset.output_types())
54 return dataset_types, dataset_shapes
63 dataset_types, dataset_shapes = _get_types_and_shapes(exec_dataset)
67 dataset_types,
Ddataset_helper.py115 self.dataset_types, self.dataset_shapes = _get_types_and_shapes(dataset)
128 return self.dataset_types, self.dataset_shapes
189 self.op = GetNextSingleOp(self.dataset_types, self.dataset_shapes, queue_name)
/third_party/mindspore/tests/ut/python/pipeline/infer/
Dinfer.py52 dataset_types = np.float32
56 np_types=dataset_types,
/third_party/mindspore/tests/ut/python/optimizer/
Dtest_optimizer_with_loss_scale.py35 def __init__(self, dataset_types, dataset_shapes): argument
37 np_types=dataset_types,
183 dataset_types = (np.float32, np.float32)
186 dataset = MindDataSet(dataset_types, dataset_shapes)
197 dataset_types = (np.float32, np.float32)
200 dataset = MindDataSet(dataset_types, dataset_shapes)
/third_party/mindspore/tests/st/pynative/loss_scale/
Dtest_loss_scale.py101 def __init__(self, dataset_types, dataset_shapes): argument
103 np_types=dataset_types,
201 dataset_types = (np.float32, np.float32)
203 dataset = MindDataSet(dataset_types, dataset_shapes)
/third_party/mindspore/tests/ut/python/communication/
Dtest_data_parallel_resnet.py277 dataset_types = (np.float32, np.float32)
283 np_types=dataset_types,
301 dataset_types = (np.float32, np.float32)
305 np_types=dataset_types,
/third_party/mindspore/tests/ut/python/train/summary/
Dtest_image_summary.py134 dataset_types = (np.float32, np.float32)
138 np_types=dataset_types,
/third_party/mindspore/tests/mindspore_test_framework/apps/
Dtest_bert_parts.py70 dataset_types = (np.int32, np.int32, np.int32, np.int32, np.int32, np.int32, np.int32)
75 np_types=dataset_types,
/third_party/mindspore/mindspore/nn/wrap/
Dcell_wrapper.py395 def __init__(self, dataset_types, dataset_shapes, queue_name): argument
397 self.get_next = P.GetNext(dataset_types, dataset_shapes, len(dataset_types), queue_name)
/third_party/mindspore/tests/ut/python/ir/
Dtest_row_tensor.py78 def __init__(self, dataset_types, dataset_shapes): argument
80 np_types=dataset_types,
354 dataset_types = (np.float32, np.float32)
356 dataset = MindDataSet(dataset_types, dataset_shapes)
/third_party/mindspore/tests/perf_test/bert/
Dtest_bert_train.py39 dataset_types = (np.int32, np.int32, np.int32, np.int32, np.int32, np.int32, np.int32)
44 np_types=dataset_types,
/third_party/mindspore/mindspore/common/
Dapi.py470 def init_dataset(self, queue_name, dataset_size, batch_size, dataset_types, dataset_shapes, argument
490 types=dataset_types,