Home
last modified time | relevance | path

Searched refs:np_types (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/tests/ut/python/communication/
Dtest_data_parallel_resnet.py246 np_types=None, output_shapes=None, input_indexs=()): argument
248 np_types=np_types, output_shapes=output_shapes,
283 np_types=dataset_types,
305 np_types=dataset_types,
/third_party/mindspore/tests/
Ddataset_mock.py25 np_types=None, output_shapes=None, input_indexs=()): argument
29 self._np_types = np_types
/third_party/mindspore/tests/ut/python/train/
Dtest_training.py86 np_types=dataset_types,
140 np_types=dataset_types,
170 np_types=dataset_types,
259 np_types=dataset_types,
Dtest_dataset_helper.py30 dataset = MindData(size=2, batch_size=batch_size, np_types=dataset_types,
Dtest_amp.py99 np_types=dataset_types,
/third_party/mindspore/tests/st/ops/ascend/
Dtest_tdt_data_ms.py77 np_types = data_set.output_types() variable
78 dataset_types = convert_type(dataset_shapes, np_types)
/third_party/mindspore/tests/ut/python/dataset/
Dtest_datasets_generator.py207np_types = [np.int8, np.int16, np.int32, np.int64, np.uint8, np.uint16, np.uint32, np.uint64, np.f…
212 for i, _ in enumerate(np_types):
213 type_tester_with_type_check(np_types[i], de_types[i])
242np_types = [np.int8, np.int16, np.int32, np.int64, np.uint8, np.uint16, np.uint32, np.uint64, np.f…
247 for i, _ in enumerate(np_types):
248 type_tester_with_type_check_2c(np_types[i], [None, de_types[i]])
762np_types = [np.int8, np.int16, np.int32, np.int64, np.uint8, np.uint16, np.uint32, np.uint64, np.f…
767 for i, _ in enumerate(np_types):
768 type_tester_with_type_check_2c_schema(np_types[i], [de_types[i], de_types[i]])
/third_party/mindspore/tests/st/data_transfer/
Dtest_tdt_data_transfer.py89 np_types = dataset.output_types()
90 dataset_types = convert_type(dataset_shapes, np_types)
/third_party/mindspore/tests/st/pynative/loss_scale/
Dtest_loss_scale.py38 np_types=None, output_shapes=None, input_indexes=(), func_name=''): argument
42 self._np_types = np_types
103 np_types=dataset_types,
/third_party/mindspore/tests/ut/python/pipeline/infer/
Dinfer.py56 np_types=dataset_types,
/third_party/mindspore/tests/ut/python/parallel/
Dtest_auto_parallel_flag.py34 np_types=dataset_types,
/third_party/mindspore/tests/ut/python/train/summary/
Dtest_image_summary.py138 np_types=dataset_types,
/third_party/mindspore/tests/mindspore_test_framework/apps/
Dtest_bert_parts.py75 np_types=dataset_types,
/third_party/mindspore/tests/perf_test/bert/
Dtest_bert_train.py44 np_types=dataset_types,
/third_party/mindspore/tests/ut/python/optimizer/
Dtest_optimizer_with_loss_scale.py37 np_types=dataset_types,
/third_party/mindspore/mindspore/common/
Dtensor.py28 np_types = (np.int8, np.int16, np.int32, np.int64, variable
82 if isinstance(input_data, np_types):
/third_party/mindspore/tests/ut/python/ir/
Dtest_row_tensor.py80 np_types=dataset_types,