Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/python/parallel/
Dtest_dataset.py21 from mindspore.ops.operations.comm_ops import _VirtualDataset
27 self.virtual_dataset = _VirtualDataset()
Dtest_auto_parallel_tuple_depend.py24 from mindspore.ops.operations.comm_ops import _VirtualDataset
61 self.virtual_dataset = _VirtualDataset()
Dtest_virtual_dataset_with_strategy.py24 from mindspore.ops.operations.comm_ops import _VirtualDataset
51 self.virtual_dataset = _VirtualDataset()
65 self.virtual_dataset = _VirtualDataset()
Dtest_unsortedsegmentmax.py25 from mindspore.ops.operations.comm_ops import _VirtualDataset
37 self.virtual_dataset = _VirtualDataset()
Dtest_unsortedsegmentmin.py25 from mindspore.ops.operations.comm_ops import _VirtualDataset
37 self.virtual_dataset = _VirtualDataset()
/third_party/mindspore/mindspore/nn/wrap/
Dcell_wrapper.py28 from ...ops.operations.comm_ops import _VirtualDataset
425 self._virtual_dataset = _VirtualDataset()
549 self._virtual_dataset = _VirtualDataset()
/third_party/mindspore/mindspore/ops/operations/
Dcomm_ops.py809 class _VirtualDataset(PrimitiveWithInfer): class
827 virtual_dataset = _VirtualDataset()
D__init__.py39 … _MirrorOperator, _MirrorMiniStepOperator, _MiniStepAllGather, ReduceOp, _VirtualDataset,