Skip to content

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