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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
REGISTER_KERNEL(ADD, Add)
REGISTER_KERNEL(ARG_MAX, ArgMax)
REGISTER_KERNEL(AVERAGE_POOL_2D, AveragePool2D)
REGISTER_KERNEL(BATCH_MATMUL, BatchMatMul)
REGISTER_KERNEL(BATCH_TO_SPACE_ND, BatchToSpaceND)
REGISTER_KERNEL(CAST, Cast)
REGISTER_KERNEL(CONCATENATION, Concatenation)
REGISTER_KERNEL(CONV_2D, Conv2D)
REGISTER_KERNEL(DEPTH_TO_SPACE, DepthToSpace)
REGISTER_KERNEL(DEPTHWISE_CONV_2D, DepthwiseConv2D)
REGISTER_KERNEL(DEQUANTIZE, Dequantize)
REGISTER_KERNEL(DIV, Div)
REGISTER_KERNEL(ELU, Elu)
REGISTER_KERNEL(EXP, Exp)
REGISTER_KERNEL(EXPAND_DIMS, ExpandDims)
REGISTER_KERNEL(FILL, Fill)
REGISTER_KERNEL(FLOOR, Floor)
REGISTER_KERNEL(FLOOR_DIV, FloorDiv)
REGISTER_KERNEL(EQUAL, Equal)
REGISTER_KERNEL(FULLY_CONNECTED, FullyConnected)
REGISTER_KERNEL(GATHER, Gather)
REGISTER_KERNEL(GREATER, Greater)
REGISTER_KERNEL(GREATER_EQUAL, GreaterEqual)
REGISTER_KERNEL(IF, If)
REGISTER_KERNEL(INSTANCE_NORM, InstanceNorm)
REGISTER_KERNEL(L2_NORMALIZATION, L2Normalize)
REGISTER_KERNEL(L2_POOL_2D, L2Pool2D)
REGISTER_KERNEL(LEAKY_RELU, LeakyRelu)
REGISTER_KERNEL(LESS, Less)
REGISTER_KERNEL(LESS_EQUAL, LessEqual)
REGISTER_KERNEL(LOCAL_RESPONSE_NORMALIZATION, LocalResponseNormalization)
REGISTER_KERNEL(LOGICAL_AND, LogicalAnd)
REGISTER_KERNEL(LOGICAL_NOT, LogicalNot)
REGISTER_KERNEL(LOGICAL_OR, LogicalOr)
REGISTER_KERNEL(LOGISTIC, Logistic)
REGISTER_KERNEL(LOG_SOFTMAX, LogSoftmax)
REGISTER_KERNEL(MAXIMUM, Maximum)
REGISTER_KERNEL(MAX_POOL_2D, MaxPool2D)
REGISTER_KERNEL(MEAN, Mean)
REGISTER_KERNEL(MINIMUM, Minimum)
REGISTER_KERNEL(MIRROR_PAD, MirrorPad)
REGISTER_KERNEL(MUL, Mul)
REGISTER_KERNEL(NEG, Neg)
REGISTER_KERNEL(NOT_EQUAL, NotEqual)
REGISTER_KERNEL(ONE_HOT, OneHot)
REGISTER_KERNEL(PACK, Pack)
REGISTER_KERNEL(PAD, Pad)
REGISTER_KERNEL(PADV2, PadV2)
REGISTER_KERNEL(POW, Pow)
REGISTER_KERNEL(PRELU, PRelu)
REGISTER_KERNEL(QUANTIZE, Quantize)
REGISTER_KERNEL(RELU, Relu)
REGISTER_KERNEL(RELU6, Relu6)
REGISTER_KERNEL(RESHAPE, Reshape)
REGISTER_KERNEL(RESIZE_BILINEAR, ResizeBilinear)
REGISTER_KERNEL(RESIZE_NEAREST_NEIGHBOR, ResizeNearestNeighbor)
REGISTER_KERNEL(REVERSE_V2, ReverseV2)
REGISTER_KERNEL(RSQRT, Rsqrt)
REGISTER_KERNEL(SHAPE, Shape)
REGISTER_KERNEL(SLICE, Slice)
REGISTER_KERNEL(SOFTMAX, Softmax)
REGISTER_KERNEL(SPACE_TO_BATCH_ND, SpaceToBatchND)
REGISTER_KERNEL(SPACE_TO_DEPTH, SpaceToDepth)
REGISTER_KERNEL(SPLIT, Split)
REGISTER_KERNEL(SPLIT_V, SplitV)
REGISTER_KERNEL(STRIDED_SLICE, StridedSlice)
REGISTER_KERNEL(SQRT, Sqrt)
REGISTER_KERNEL(SQUARE, Square)
REGISTER_KERNEL(SQUARED_DIFFERENCE, SquaredDifference)
REGISTER_KERNEL(SQUEEZE, Squeeze)
REGISTER_KERNEL(SUB, Sub)
REGISTER_KERNEL(SVDF, SVDF)
REGISTER_KERNEL(TANH, Tanh)
REGISTER_KERNEL(TRANSPOSE, Transpose)
REGISTER_KERNEL(TRANSPOSE_CONV, TransposeConv)
REGISTER_KERNEL(UNPACK, Unpack)
REGISTER_KERNEL(WHILE, While)
|
