-
Notifications
You must be signed in to change notification settings - Fork 1
/
raw_ops.py
28 lines (28 loc) · 1.1 KB
/
raw_ops.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from .array_ops import (
Concat, ConcatOffset, ExpandDims, Fill,
InvertPermutation, ListDiff, Pack, Pad, Range, Reshape,
Slice, Squeeze, StridedSlice, StridedSliceGrad, Tile,
Transpose, Unpack,
)
from .data_flow_ops import (
BroadcastTo, DynamicStitch, Gather, Identity,
Select, StopGradient,
)
from .generic_ops import Const, OnesLike, Rank, Shape, Size, ZerosLike
from .math_ops import (
Add, AddN, BatchMatMul, BroadcastGradientArgs, Cumprod,
Cumsum, DivNoNan, Equal, Exp, FloorDiv, FloorMod,
Greater, GreaterEqual, Less, LessEqual, Log, Log1p,
MatMul, Maximum, Mean, Minimum, Mul, Neg, NotEqual,
Prod, RealDiv, Reciprocal, ReciprocalGrad, Rsqrt,
RsqrtGrad, Sqrt, SqrtGrad, Square, SquaredDifference,
Sub, Sum,
)
from .nn_ops import (
AvgPool2D, AvgPool2DGrad, Conv2D, Conv2DBackpropFilter,
Conv2DBackpropInput, LeakyRelu, LeakyReluGrad, LogSoftmax,
MaxPool2D, MaxPool2DGrad, MaxPool2DGradGrad, Relu,
ReluGrad, Sigmoid, SigmoidGrad, Softmax,
SoftmaxCrossEntropyWithLogits, Tanh, TanhGrad,
)
from .random_ops import RandomStandardNormal, RandomUniform