Commit e5b50db7 authored by shssf's avatar shssf Committed by Scott Cyphers

IntelGPU backend: Sin, cos, log, exp operations (#1381)

parent 56529a0e
......@@ -37,6 +37,7 @@
#include "ngraph/runtime/intelgpu/intelgpu_op_custom_kernels.hpp"
#include "ngraph/runtime/intelgpu/intelgpu_tensor_view.hpp"
#include "ngraph/function.hpp"
#include "ngraph/node.hpp"
#include "ngraph/op/avg_pool.hpp"
#include "ngraph/op/batch_norm.hpp"
......@@ -50,6 +51,7 @@
#include "ngraph/op/max_pool.hpp"
#include "ngraph/op/min.hpp"
#include "ngraph/op/pad.hpp"
#include "ngraph/op/parameter_vector.hpp"
#include "ngraph/op/product.hpp"
#include "ngraph/op/reshape.hpp"
#include "ngraph/op/reverse.hpp"
......@@ -574,6 +576,38 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func)
{
do_unary_operation(topology, op, activation_hyperbolic_tan);
}
else if ("Sin" == op->description())
{
do_unary_operation(topology, op, activation_sin);
}
else if ("Asin" == op->description())
{
do_unary_operation(topology, op, activation_asin);
}
else if ("Sinh" == op->description())
{
do_unary_operation(topology, op, activation_sinh);
}
else if ("Cos" == op->description())
{
do_unary_operation(topology, op, activation_cos);
}
else if ("Acos" == op->description())
{
do_unary_operation(topology, op, activation_acos);
}
else if ("Cosh" == op->description())
{
do_unary_operation(topology, op, activation_cosh);
}
else if ("Log" == op->description())
{
do_unary_operation(topology, op, activation_log);
}
else if ("Exp" == op->description())
{
do_unary_operation(topology, op, activation_exp);
}
else if ("Sigmoid" == op->description())
{
do_unary_operation(topology, op, activation_logistic);
......
abc_int64
acos
aliased_output
asin
atan
avg_pool_2d_2channel_2image_padded_only_above
avg_pool_3d
backwards_abs
backwards_acos
backwards_asin
backwards_atan
backwards_avgpool_n1_c1_hw2x2
backwards_avgpool_n1_c1_hw4x4
......@@ -23,16 +19,13 @@ backwards_dot_scalar_tensor
backwards_dot_tensor3_tensor3
backwards_dot_tensor_scalar
backwards_dot_tensor_vector
backwards_exp
backwards_floor
backwards_log
backwards_maximum
backwards_maxpool_n2c1h5w5_kh3kw3_sh2sw2
backwards_maxpool_n2_c1_hw5_3x3_str2_max
backwards_maxpool_n4c1h4w4_kh2kw2_sh1sw1
backwards_maxpool_n4_c1_hw4_2x2_max
backwards_minimum
backwards_power
backwards_replace_slice
backwards_reverse_sequence_n3_c2_h3
backwards_reverse_sequence_n4d2c3h2w2
......@@ -40,8 +33,6 @@ backwards_select
backwards_select_nested
backwards_sigmoid
backwards_sign
backwards_sin
backwards_sinh
backwards_slice
backwards_softmax_3d
backwards_softmax_all
......@@ -90,14 +81,10 @@ convolution_4d_4items_strided_dilated_padded
convolution_4d_4items_strided_dilated_padded_neg
convolution_4d_4items_strided_dilated_padded_same
convolution_outlining
cos
cosh
divide_by_zero_int32
dot_matrix_vector_int64
exp
floor
function_call
log
lrn
max_pool_3d
minimum
......@@ -146,8 +133,6 @@ select_and_scatter_without_overlap
select_and_scatter_with_overlap
sigmoid_bprop_n1c1h4
sign
sin
sinh
softmax_all
softmax_axis
softmax_axis_2
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment