The following table lists the supported operations and APIs for vai_q_tensorflow.
| Type | Operation Type | tf.nn | tf.layers | tf.keras.layers |
|---|---|---|---|---|
| Convolution | Conv2D DepthwiseConv2dNative |
atrous_conv2d conv2d conv2d_transpose depthwise_conv2d_native separable_conv2d |
Conv2D Conv2DTranspose SeparableConv2D
|
Conv2D Conv2DTranspose DepthwiseConv2D SeparaleConv2D |
| Fully Connected | MatMul | / | Dense | Dense |
| BiasAdd | BiasAdd Add |
bias_add | / | / |
| Pooling | AvgPool Mean MaxPool |
avg_pool max_pool |
AveragePooling2D MaxPooling2D |
AveragePooling2D MaxPool2D |
| Activation | ReLU ReLU6 Sigmoid Swish Hard-sigmoid Hard-swish |
relu relu6 leaky_relu swish |
/ | ReLU Leaky ReLU |
| BatchNorm[#1] | FusedBatchNorm | batch_normalization batch_norm_with_global_normalization fused_batch_norm |
BatchNormalization | BatchNormalization |
| Upsampling | ResizeBilinear ResizeNearestNeighbor |
/ | / | UpSampling2D |
| Concat | Concat ConcatV2 |
/ | / | Concatenate |
| Others | Placeholder Const Pad Squeeze Reshape ExpandDims |
dropout[#2] softmax[#3] |
Dropout[#2] Flatten |
Input Flatten Reshape Zeropadding2D Softmax |
|
||||