Signed-off-by: Kainan Cha <kainan.zha@verisilicon.com> |
||
|---|---|---|
| .. | ||
| README.md | ||
| activations.cc | ||
| addn.cc | ||
| arg.cc | ||
| batch2space.cc | ||
| batchnorm.cc | ||
| clip.cc | ||
| concat.cc | ||
| conv2d.cc | ||
| deconv.cc | ||
| deconv2d_test.cc | ||
| depth2space.cc | ||
| dropout.cc | ||
| elementwise.cc | ||
| fullyconnected.cc | ||
| gather.cc | ||
| gathernd.cc | ||
| l2normalization.cc | ||
| localresponsenormalization.cc | ||
| logical.cc | ||
| nbg.cc | ||
| pad.cc | ||
| pool2d.cc | ||
| reduce.cc | ||
| relational_operations.cc | ||
| relational_operations_test.cc | ||
| reorg.cc | ||
| reorg_test.cc | ||
| reshape.cc | ||
| resize.cc | ||
| reverse.cc | ||
| select.cc | ||
| simple_operations.cc | ||
| simple_operations_test.cc | ||
| slice.cc | ||
| softmax.cc | ||
| space2batch.cc | ||
| space2depth.cc | ||
| split.cc | ||
| squeeze.cc | ||
| stack.cc | ||
| stridedslice.cc | ||
| transpose.cc | ||
README.md
INFO: Actual implementations may differ from reference link in terms of dimensions and parameters supported
| TIM-VX API | Internal Op | Status | Reference |
|---|---|---|---|
| Add | ADD | Mapped | tf.math.add |
| Multiply | MULTIPLY | Mapped | tf.math.multiply |
| Conv2d | CONV2D | Mapped | tf.nn.conv2d tf.nn.atros_conv2d tf.nn.depthwise_conv2d |
| Softmax | SOFTMAX | Mapped | tf.nn.softmax |
| Pool2d | POOL | Mapped | tf.nn.pool |
| LeakyRelu | LEAKY_RELU | Mapped | tf.nn.leaky_relu |
| Concat | CONCAT | Mapped | tf.concat |
| Split | SPLIT | Mapped | tf.split |
| NOOP | Unmapped | ||
| ROI_POOL | Unmapped | ANEURALNETWORKS_ROI_POOLING | |
| BatchNorm | BATCH_NORM | Mapped | tf.nn.batch_normalization |
| PROPOSAL | Unmapped | ||
| DeConv2d | DECONVOLUTION | Mapped | tf.nn.conv2d_transpose |
| Reshape | RESHAPE | Mapped | tf.reshape |
| Transpose | PERMUTE | Mapped | tf.transpose |
| Prelu | PRELU | Mapped | tf.keras.layers.PReLU |
| UPSAMPLE | Unmapped | ||
| Relu | RELU | Mapped | tf.nn.relu |
| RELUN | Deprecated | tf.keras.layers.ReLU(max_value=N) | |
| LSTM | Unmapped | tf.keras.layers.LSTM | |
| Reorg | REORG | Mapped | darknet.reorg |
| VARIABLE | Unmapped | tf.variable | |
| L2Normalization | L2_NORMALIZE | Mapped | tf.math.l2_normalize |
| FullyConnected | FCL2 | Mapped | tf.keras.layers.Dense |
| POOLWITHARGMAX | Unmapped | tf.nn.max_pool_with_argmax | |
| ArgMax | ARGMAX | Mapped | tf.math.argmax |
| Maximum | MAXIMUM | Mapped | tf.math.maximum |
| L2NORMALIZESCALE | UnMapped | ||
| CROP | Unmapped | ||
| Sub | SUBTRACT | Mapped | tf.math.subtract |
| Relu6 | RELU6 | Mapped | tf.nn.relu6 |
| Sigmoid | SIGMOID | Mapped | tf.math.sigmoid |
| Tanh | TANH | Mapped | tf.math.tanh |
| Sqrt | SQRT | Mapped | tf.math.sqrt |
| Rsqrt | RSQRT | Mapped | tf.math.rsqrt |
| SoftRelu | SOFTRELU | Mapped | tf.math.softplus |
| Div | DIVIDE | Mapped | tf.math.divide |
| Dropout | DROPOUT | Mapped | f(x) = x*ratio |
| SHUFFLECHANNEL | Unmapped | ANEURALNETWORKS_CHANNEL_SHUFFLE | |
| Resize | RESIZE | Mapped | tf.image.resize |
| Reverse | REVERSE | Mapped | tf.reverse |
| DepthToSpace | DEPTH2SPACE | Mapped | tf.nn.depth_to_space |
| SpaceToDepth | SPACE2DEPTH | Mapped | tf.nn.space_to_depth |
| DataConvert | DATACONVERT | Mapped | |
| Slice | SLICE | Mapped | tf.slice |
| Elu | ELU | Mapped | tf.nn.elu |
| Batch2Space | BATCH2SPACE | Mapped | tf.batch_to_space |
| Space2Batch | SPACE2BATCH | Mapped | tf.space_to_batch |
| Pad | PAD | Mapped | tf.pad |
| IMAGEPROCESS | Unmapped | ||
| MATRIXMUL | Unmapped | tf.experimental.numpy.matmul | |
| LSTMUNIT | Unmapped | ||
| LAYER_NORM | Unmapped | tf.keras.layers.LayerNormalization | |
| ReduceMin | REDUCE_MIN | Mapped | tf.math.reduce_min |
| ReduceMax | REDUCE_MAX | Mapped | tf.math.reduce_max |
| ReduceAny | REDUCE_ANY | Mapped | tf.math.reduce_any |
| ReduceProd | REDUCE_PROD | Mapped | tf.math.reduce_prod |
| ReduceMean | REDUCE_MEAN | Mapped | tf.math.reduce_mean |
| INSTANCE_NORM | Unmapped | tfa.layers.InstanceNormalization | |
| TENSORSTACKCONCAT | Unmapped | ||
| StridedSlice | STRIDED_SLICE | Mapped | tf.strided_slice |
| SIGNAL_FRAME | Unmapped | ||
| A_TIMES_B_PLUS_C | Unmapped | tf.add(tf.mul(A, B), C) | |
| SVDF | Unmapped | ANEURALNETWORKS_SVDF | |
| Abs | ABS | Mapped | tf.math.abs |
| CONV1D | Unmapped | tf.nn.conv1d | |
| NBG | NBG | Mapped | |
| CONCATSHIFT | Unmapped | ||
| LocalResponseNormalization | LRN2 | Mapped | tf.nn.local_response_normalization |
| Greater | RELATIONAL_OPS_GREATER | Mapped | tf.math.greater |
| GreaterOrEqual | RELATIONAL_OPS_GREATER_EQUAL | Mapped | tf.math.greater_equal |
| Less | RELATIONAL_OPS_LESS | Mapped | tf.math.less |
| LessOrEqual | RELATIONAL_OPS_LESS_EQUAL | Mapped | tf.math.less_equal |
| Equal | RELATIONAL_OPS_EQUAL | Mapped | tf.math.equal |
| NotEqual | RELATIONAL_OPS_NOT_EQUAL | Mapped | tf.math.not_equal |
| SYNC_HOST | Unmapped | ||
| Pow | POW | Mapped | tf.math.pow |
| FLOORDIV | Unmapped | tf.math.floordiv | |
| Minimum | MINIMUM | Mapped | tf.math.minimum |
| SPATIAL_TRANSFORMER | Unmapped | ||
| And | LOGICAL_OPS | Mapped | tf.math.logical_and |
| Or | LOGICAL_OPS | Mapped | tf.math.logical_or |
| Select | SELECT | Mapped | tf.where |
| LSTMUNIT_ACTIVATION | Unmapped | ||
| LSTMUNIT_OVXLIB | Unmapped | ||
| TENSOR_ADD_MEAN_STDDEV_NORM | Unmapped | ||
| Relu1 | RELU1 | Mapped | tf.keras.layers.ReLU(max_value=1.0) |
| Stack | STACK | Mapped | tf.stack |
| Floor | FLOOR | Mapped | tf.math.floor |
| Square | SQUARE | Mapped | tf.math.square |
| Neg | NEG | Mapped | tf.math.negative |
| Exp | EXP | Mapped | tf.math.exp |
| LSTM_OVXLIB | Unmapped | ||
| PRE_PROCESS_TENSOR | Unmapped | ||
| HASHTABLE_LOOKUP | Unmapped | ANEURALNETWORKS_HASHTABLE_LOOKUP | |
| EMBEDDING_LOOKUP | Unmapped | ANEURALNETWORKS_EMBEDDING_LOOKUP | |
| LSH_PROJECTION | Unmapped | ANEURALNETWORKS_LSH_PROJECTION | |
| RNN | Unmapped | ANUERALNETWORKS_RNN | |
| Clip | CLIP | Mapped | tf.clip_by_value |
| POST_PROCESS | Unmapped | ||
| PRE_PROCESS_GRAY | Unmapped | ||
| UNSTACK | Unmapped | tf.unstack | |
| PRE_PROCESS_RGB | Unmapped | ||
| PRE_PROCESS | Unmapped | ||
| AddN | ADDN | Mapped | tf.math.add_n |
| PRE_PROCESS_YUV420 | Unmapped | ||
| EXTRA_ENDING | InternalOnly | ||
| Gather | GATHER | Mapped | tf.gather |
| TILE | Unmapped | tf.tile | |
| GROUPED_CONV2D | Unmapped | ANEURALNETWORKS_GROUPED_CONV_2D | |
| TOPK | Unmapped | tf.math.top_k | |
| PRE_PROCESS_BGRA | Unmapped | ||
| LogicalNot | LOGICAL_NOT | Mapped | tf.math.logical_not |
| Sin | SIN | Mapped | tf.math.sin |
| Log | LOG | Mapped | tf.math.log |
| ArgMin | ARGMIN | Mapped | tf.math.argmin |
| ROI_ALIGN | Unmapped | ANEURALNETWORKS_ROI_ALIGN | |
| HEATMAP_MAX_KEYPOINT | Unmapped | ANEURALNETWORKS_HEATMAP_MAX_KEYPOINT | |
| AXIS_ALIGNED_BBOX_TRANSFORM | Unmapped | ANEURALNETWORKS_AXIS_ALIGNED_BBOX_TRANSFORM | |
| BOX_WITH_NMS_LIMIT | Unmapped | ANEURALNETWORKS_BOX_WITH_NMX_LIMIT | |
| GENERATE_PROPOSALS | Unmapped | ANEURALNETWORKS_GENERATE_PROPOSALS | |
| DETECTION_POSTPROCESS | Unmapped | ANEURALNETWORKS_DETECTION_POSTPROCESSING | |
| RANDOM_MULTINOMIAL | Unmapped | ANEURALNETWORKS_RANDOM_MULTINOMIAL | |
| LOG_SOFTMAX | Unmapped | tf.nn.log_softmax | |
| RELU_KERAS | Unmapped | tf.keras.layers.ReLU | |
| GRU_OVXLIB | Unmapped | ||
| GRUCELL_OVXLIB | Unmapped | ||
| UNIDIRECTIONAL_SEQUENCE_RNN | Unmapped | ||
| QUANTIZED_16BIT_LSTM | Unmapped | ||
| BIDIRECTIONAL_SEQUENCE_RNN | Unmapped | ||
| BIDIRECTIONAL_SEQUENCE_LSTM | Unmapped | ||
| RNNCELL_OVXLIB | Unmapped | ||
| HardSwish | SWISH | Mapped | tf.keras.activations.swish |
| DEPTHWISE_CONV1D | Unmapped | ||
| GatherNd | GATHER_ND | Mapped | tf.gather_nd |
| Cast | CAST | Mapped | tf.cast |
| LINEAR | Unmapped | f(x) = a*x + b | |
| BATCHNORM_SINGLE | Unmapped | tf.nn.batch_normalization | |
| MOMENTS | Unmapped | tf.moments | |
| Squeeze | SQUEEZE | Mapped | tf.squeeze |
| HardSigmoid | HARD_SIGMOID | Mapped | tf.keras.activations.hard_sigmoid |
| Mish | MISH | Mapped | tfa.activations.mish |
| EXPAND_BROADCAST | Unmapped | ||
| PRE_PROCESS_YUV444 | Unmapped | ||
| PRE_PROCESS_NV12 | Unmapped | ||
| SCATTER_ND | Unmapped | tf.scatter_nd | |
| DECONVOLUTION1D | Unmapped | tf.nn.conv1d_transpose | |
| INTERP | Unmapped | ||
| RESIZE_1D | Unmapped | ||
| CONV_RELU | Deprecated | ||
| CONV_RELU_POOL | Deprecated | ||
| FCL | Deprecated | ||
| FCL_RELU | Deprecated | ||
| LRN | Deprecated | ||
| SCALE | Deprecated |