|
|
|
@ -82,7 +82,8 @@ from .nn_ops import (LSTM, SGD, Adam, FusedSparseAdam, FusedSparseLazyAdam, Appl
|
|
|
|
|
ApplyRMSProp, ApplyCenteredRMSProp, BasicLSTMCell, InTopK, UniformCandidateSampler)
|
|
|
|
|
from . import _quant_ops
|
|
|
|
|
from ._quant_ops import *
|
|
|
|
|
from .other_ops import (Assign, InplaceAssign, IOU, BoundingBoxDecode, BoundingBoxEncode, PopulationCount,
|
|
|
|
|
from .other_ops import (Assign, InplaceAssign, IOU, BoundingBoxDecode, BoundingBoxEncode,
|
|
|
|
|
ConfusionMatrix, PopulationCount,
|
|
|
|
|
CheckValid, MakeRefKey, Partial, Depend, identity, CheckBprop, Push, Pull)
|
|
|
|
|
from ._thor_ops import (CusBatchMatMul, CusCholeskyTrsm, CusFusedAbsMax1, CusImg2Col, CusMatMulCubeDenseLeft,
|
|
|
|
|
CusMatMulCubeFraczRightMul, CusMatMulCube, CusMatrixCombine, CusTranspose02314,
|
|
|
|
@ -289,6 +290,7 @@ __all__ = [
|
|
|
|
|
'DepthwiseConv2dNative',
|
|
|
|
|
'UnsortedSegmentSum',
|
|
|
|
|
'UnsortedSegmentMin',
|
|
|
|
|
'UnsortedSegmentMax',
|
|
|
|
|
'UnsortedSegmentProd',
|
|
|
|
|
"AllGather",
|
|
|
|
|
"AllReduce",
|
|
|
|
@ -377,6 +379,7 @@ __all__ = [
|
|
|
|
|
"UniformCandidateSampler",
|
|
|
|
|
"LRN",
|
|
|
|
|
"Mod",
|
|
|
|
|
"ConfusionMatrix",
|
|
|
|
|
"PopulationCount",
|
|
|
|
|
"ParallelConcat",
|
|
|
|
|
"Push",
|
|
|
|
|