Searched refs:virtual_dataset (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/parallel/ |
| D | virtual_dataset_test.cc | 30 VirtualDatasetInfoPtr virtual_dataset; variable
61 …virtual_dataset = std::make_shared<VirtualDatasetInfo>("virtual_dataset_info", inputs_shape, outpu… in SetUp()
67 virtual_dataset->Init(strategy); in TEST_F()
68 Shape dev_matrix_shape = virtual_dataset->dev_matrix_shape(); in TEST_F()
78 virtual_dataset->Init(strategy); in TEST_F()
79 OperatorVector forward_op = virtual_dataset->forward_op(); in TEST_F()
89 virtual_dataset->Init(strategy); in TEST_F()
90 MirrorOps mirror_ops = virtual_dataset->mirror_ops(); in TEST_F()
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_dataset.py | 27 self.virtual_dataset = _VirtualDataset()
33 x, y, z = self.virtual_dataset(x, y, z)
|
| D | test_virtual_dataset_with_strategy.py | 51 self.virtual_dataset = _VirtualDataset()
57 x, y, b = self.virtual_dataset(x, y, b)
65 self.virtual_dataset = _VirtualDataset()
73 x, y, b = self.virtual_dataset(x, y, b)
|
| D | test_auto_parallel_tuple_depend.py | 61 self.virtual_dataset = _VirtualDataset()
68 x, y, b = self.virtual_dataset(x, y, b)
|
| D | test_unsortedsegmentmax.py | 37 self.virtual_dataset = _VirtualDataset()
|
| D | test_unsortedsegmentmin.py | 37 self.virtual_dataset = _VirtualDataset()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pass.cc | 329 opt::OptPassConfig virtual_dataset = opt::OptPassConfig({irpass.virtual_dataset_eliminate_}); in GetOptPassesA() local
348 {"virtual_dataset", virtual_dataset}, in GetOptPassesA()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | comm_ops.py | 827 virtual_dataset = _VirtualDataset() variable
|