-
Notifications
You must be signed in to change notification settings - Fork 3
Summary of operators in OneFlow XRT
Houjiang Chen edited this page Jun 9, 2022
·
3 revisions
OP | XLA | TENSORRT | OPENVINO |
---|---|---|---|
adam_update | ✔ | ||
add_n | ✔ | ✔ | ✔ |
batch_gather | ✔ | ||
bias_add | ✔ | ✔ | |
broadcast_add | ✔ | ✔ | |
broadcast_div | ✔ | ✔ | |
broadcast_div_grad | ✔ | ||
broadcast_like | ✔ | ||
broadcast_min | |||
broadcast_mul | ✔ | ✔ | |
broadcast_sub | ✔ | ✔ | |
cast | ✔ | ✔ | |
concat | ✔ | ✔ | |
conv2d | ✔ | ✔ | |
conv_data_grad | ✔ | ||
deconv2d | ✔ | ||
gather | ✔ | ||
gelu | ✔ | ||
gelu_grad | ✔ | ||
identity | ✔ | ✔ | |
layer_norm | ✔ | ||
layer_norm_grad | ✔ | ||
layer_norm_param_grad | ✔ | ||
leaky_relu | ✔ | ✔ | |
matmul | ✔ | ✔ | ✔ |
multiply | ✔ | ||
normalization | ✔ | ✔ | |
pad | ✔ | ||
pad_grad | ✔ | ||
prelu | ✔ | ||
prelu_grad | ✔ | ||
reduce_mean | ✔ | ||
reduce_sum | ✔ | ✔ | |
relu | ✔ | ✔ | |
reshape | ✔ | ✔ | ✔ |
reshape_like | ✔ | ✔ | |
rsqrt | ✔ | ✔ | |
scalar_add | ✔ | ✔ | |
scalar_div | ✔ | ✔ | |
scalar_mul | ✔ | ✔ | |
scalar_pow | ✔ | ✔ | |
scalar_pow_grad | ✔ | ||
scalar_sub | ✔ | ||
sigmoid | ✔ | ✔ | |
sigmoid_v2 | ✔ | ||
softmax | ✔ | ✔ | |
softmax_grad | ✔ | ||
sqrt | ✔ | ✔ | |
sqrt_grad | ✔ | ||
square_sum | ✔ | ||
tanh | ✔ | ✔ | |
tanh_grad | ✔ | ||
tf_avg_pool_2d | ✔ | ||
tf_max_pool_2d | ✔ | ||
top_k | ✔ | ||
transpose | ✔ | ✔ | |
upsample_nearest | ✔ |