operators.hpp 3.74 KB
Newer Older
Paul's avatar
Paul committed
1
2
#ifndef MIGRAPHX_GUARD_OPERATORS_HPP
#define MIGRAPHX_GUARD_OPERATORS_HPP
Paul's avatar
Paul committed
3

4
5
#include <migraphx/op/abs.hpp>
#include <migraphx/op/acos.hpp>
6
#include <migraphx/op/acosh.hpp>
7
#include <migraphx/op/add.hpp>
8
9
#include <migraphx/op/argmax.hpp>
#include <migraphx/op/argmin.hpp>
10
#include <migraphx/op/asin.hpp>
11
#include <migraphx/op/asinh.hpp>
12
13
#include <migraphx/op/as_shape.hpp>
#include <migraphx/op/atan.hpp>
14
#include <migraphx/op/atanh.hpp>
15
#include <migraphx/op/batch_norm_inference.hpp>
16
17
#include <migraphx/op/binary.hpp>
#include <migraphx/op/broadcast.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
18
#include <migraphx/op/capture.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
19
#include <migraphx/op/ceil.hpp>
Khalique's avatar
Khalique committed
20
#include <migraphx/op/clip.hpp>
21
22
23
#include <migraphx/op/common.hpp>
#include <migraphx/op/concat.hpp>
#include <migraphx/op/contiguous.hpp>
24
#include <migraphx/op/convert.hpp>
25
26
27
#include <migraphx/op/convolution.hpp>
#include <migraphx/op/cosh.hpp>
#include <migraphx/op/cos.hpp>
kahmed10's avatar
kahmed10 committed
28
#include <migraphx/op/deconvolution.hpp>
29
30
31
#include <migraphx/op/div.hpp>
#include <migraphx/op/dot.hpp>
#include <migraphx/op/elu.hpp>
32
#include <migraphx/op/equal.hpp>
33
#include <migraphx/op/erf.hpp>
34
35
#include <migraphx/op/exp.hpp>
#include <migraphx/op/flatten.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
36
#include <migraphx/op/floor.hpp>
37
#include <migraphx/op/gather.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
38
#include <migraphx/op/get_tuple_elem.hpp>
39
#include <migraphx/op/greater.hpp>
40
41
#include <migraphx/op/gru.hpp>
#include <migraphx/op/identity.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
42
#include <migraphx/op/if_op.hpp>
43
#include <migraphx/op/im2col.hpp>
Charlie Lin's avatar
Charlie Lin committed
44
#include <migraphx/op/isnan.hpp>
45
#include <migraphx/op/leaky_relu.hpp>
46
#include <migraphx/op/less.hpp>
47
48
#include <migraphx/op/load.hpp>
#include <migraphx/op/log.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
49
50
51
#include <migraphx/op/logical_and.hpp>
#include <migraphx/op/logical_or.hpp>
#include <migraphx/op/logical_xor.hpp>
52
#include <migraphx/op/logsoftmax.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
53
#include <migraphx/op/loop.hpp>
54
55
56
57
58
59
60
#include <migraphx/op/lrn.hpp>
#include <migraphx/op/lstm.hpp>
#include <migraphx/op/max.hpp>
#include <migraphx/op/min.hpp>
#include <migraphx/op/mul.hpp>
#include <migraphx/op/multibroadcast.hpp>
#include <migraphx/op/neg.hpp>
61
#include <migraphx/op/nonmaxsuppression.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
62
#include <migraphx/op/nonzero.hpp>
63
64
65
#include <migraphx/op/outline.hpp>
#include <migraphx/op/pad.hpp>
#include <migraphx/op/pooling.hpp>
turneram's avatar
turneram committed
66
67
#include <migraphx/op/pow.hpp>
#include <migraphx/op/prefix_scan_sum.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
68
#include <migraphx/op/prelu.hpp>
69
#include <migraphx/op/quant_convolution.hpp>
70
#include <migraphx/op/quant_dot.hpp>
71
#include <migraphx/op/recip.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
72
#include <migraphx/op/reduce_max.hpp>
73
#include <migraphx/op/reduce_mean.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
74
#include <migraphx/op/reduce_min.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
75
76
#include <migraphx/op/reduce_prod.hpp>
#include <migraphx/op/reduce_sum.hpp>
77
78
#include <migraphx/op/relu.hpp>
#include <migraphx/op/reshape.hpp>
Cagri Eryilmaz's avatar
Cagri Eryilmaz committed
79
#include <migraphx/op/reverse.hpp>
80
81
#include <migraphx/op/rnn.hpp>
#include <migraphx/op/rnn_last_cell_output.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
82
83
84
#include <migraphx/op/rnn_last_hs_output.hpp>
#include <migraphx/op/rnn_variable_seq_lens.hpp>
#include <migraphx/op/rnn_var_sl_last_output.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
85
#include <migraphx/op/roialign.hpp>
86
#include <migraphx/op/round.hpp>
Khalique's avatar
Khalique committed
87
#include <migraphx/op/rsqrt.hpp>
88
#include <migraphx/op/scalar.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
89
#include <migraphx/op/scatter.hpp>
90
#include <migraphx/op/sigmoid.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
91
#include <migraphx/op/sign.hpp>
92
93
94
95
#include <migraphx/op/sinh.hpp>
#include <migraphx/op/sin.hpp>
#include <migraphx/op/slice.hpp>
#include <migraphx/op/softmax.hpp>
96
#include <migraphx/op/sqrt.hpp>
Khalique's avatar
Khalique committed
97
#include <migraphx/op/sqdiff.hpp>
98
#include <migraphx/op/squeeze.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
99
#include <migraphx/op/step.hpp>
100
101
102
#include <migraphx/op/sub.hpp>
#include <migraphx/op/tanh.hpp>
#include <migraphx/op/tan.hpp>
Shucai Xiao's avatar
Shucai Xiao committed
103
#include <migraphx/op/topk.hpp>
104
105
#include <migraphx/op/transpose.hpp>
#include <migraphx/op/unary.hpp>
106
#include <migraphx/op/unary_not.hpp>
107
108
#include <migraphx/op/undefined.hpp>
#include <migraphx/op/unknown.hpp>
109
#include <migraphx/op/unsqueeze.hpp>
turneram's avatar
turneram committed
110
#include <migraphx/op/where.hpp>
Paul's avatar
Paul committed
111
112

#endif