Home
last modified time | relevance | path

Searched refs:PrimitiveWithCheck (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ops/operations/
D_embedding_cache_ops.py18 from ..primitive import prim_attr_register, PrimitiveWithCheck
22 class UpdateCache(PrimitiveWithCheck):
59 class SubAndFilter(PrimitiveWithCheck):
104 class MapUniform(PrimitiveWithCheck):
144 class CacheSwapTable(PrimitiveWithCheck):
183 class MapCacheIdx(PrimitiveWithCheck):
235 class DynamicAssign(PrimitiveWithCheck):
266 class PadAndShift(PrimitiveWithCheck):
D_inner_ops.py24 from ..primitive import PrimitiveWithCheck, PrimitiveWithInfer, prim_attr_register, Primitive
603 class GpuConvertToDynamicShape(PrimitiveWithCheck):
701 class SequenceMask(PrimitiveWithCheck):
1033 class DynamicStitch(PrimitiveWithCheck):
Darray_ops.py32 from ..primitive import Primitive, PrimitiveWithInfer, PrimitiveWithCheck, prim_attr_register, _run…
76 class _ScatterOpDynamic(PrimitiveWithCheck):
841 class GatherV2(PrimitiveWithCheck):
863 class SparseGatherV2(PrimitiveWithCheck):
1013 class Split(PrimitiveWithCheck):
2104 class UnsortedSegmentMin(PrimitiveWithCheck):
2174 class UnsortedSegmentMax(PrimitiveWithCheck):
5830 class EmbeddingLookup(PrimitiveWithCheck):
5980 class Range(PrimitiveWithCheck):
6084 class MaskedSelect(PrimitiveWithCheck):
Dother_ops.py24 from ..primitive import Primitive, PrimitiveWithCheck, PrimitiveWithInfer, prim_attr_register
120 class Load(PrimitiveWithCheck):
Dcomm_ops.py23 from ..primitive import PrimitiveWithInfer, PrimitiveWithCheck, Primitive, prim_attr_register
572 class AllSwap(PrimitiveWithCheck):
Dmath_ops.py22 from ..primitive import Primitive, PrimitiveWithInfer, PrimitiveWithCheck, prim_attr_register, _run…
1109 class MatMul(PrimitiveWithCheck):
1899 class Sqrt(PrimitiveWithCheck):
Dnn_ops.py30 from ..primitive import Primitive, PrimitiveWithInfer, PrimitiveWithCheck, prim_attr_register
562 class ReLU6(PrimitiveWithCheck):
900 class FusedBatchNormEx(PrimitiveWithCheck):
2963 class SGD(PrimitiveWithCheck):
4352 class ComputeAccidentalHits(PrimitiveWithCheck):
6310 class SparseApplyProximalAdagrad(PrimitiveWithCheck):
7033 class SparseApplyFtrl(PrimitiveWithCheck):
7249 class Dropout(PrimitiveWithCheck):
7494 class CTCGreedyDecoder(PrimitiveWithCheck):
/third_party/mindspore/mindspore/ops/
Dprimitive.py341 class PrimitiveWithCheck(Primitive): class
602 elif isinstance(self, PrimitiveWithCheck):
603 PrimitiveWithCheck.__init__(self, class_name)
D__init__.py24 from .primitive import Primitive, PrimitiveWithInfer, PrimitiveWithCheck, prim_attr_register