api_compat.yaml 3.5 KB
Newer Older
1 2 3 4 5
# - api : conv3d_transpose
#   backward : conv3d_transpose_grad
#   extra :
#     attrs : [bool use_cudnn = true, bool use_mkldnn = false, int workspace_size_MB = platform::GetDefaultConvWorkspaceSizeLimitMB()]

6 7
- api : atan2
  inputs :
8
    {x : X1, y : X2}
9 10 11
  outputs :
    out : Out

12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
- api : bernoulli
  inputs :
    x : X
  outputs :
    out : Out

- api : cholesky
  inputs :
    x : X
  outputs :
    out : Out

- api : cholesky_solve
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

- api : conv2d
31
  backward : conv2d_grad
32 33 34
  extra :
    attrs : [bool use_cudnn = false, bool fuse_relu_before_depthwise_conv = false, bool use_mkldnn = false,
             bool use_quantizer = false, str mkldnn_data_type = "float32", bool fuse_relu = false,
35
             str fuse_activation = "", float fuse_alpha = 0.0f, float fuse_beta = 0.0f, bool use_addto = false,
36 37 38 39
             bool fuse_residual_connection = false, float Scale_in = 1.0f, float Scale_out = 1.0f,
             float Scale_in_eltwise = 1.0f, 'float[] Scale_weights = {1.0f}', bool force_fp32_output = false,
             int workspace_size_MB = 512, bool exhaustive_search = false]

40
- api : conv2d_fusion
F
Feiyu Chan 已提交
41 42 43
  extra :
    attrs : [bool use_cudnn = false, bool fuse_relu_before_depthwise_conv = false, bool use_mkldnn = false,
             bool use_quantizer = false, str mkldnn_data_type = "float32", bool fuse_relu = false,
44
             str fuse_activation = "", float fuse_alpha = 0.0f, float fuse_beta = 0.0f, bool use_addto = false,
F
Feiyu Chan 已提交
45 46 47 48
             bool fuse_residual_connection = false, float Scale_in = 1.0f, float Scale_out = 1.0f,
             float Scale_in_eltwise = 1.0f, 'float[] Scale_weights = {1.0f}', bool force_fp32_output = false,
             int workspace_size_MB = 512, bool exhaustive_search = false]

49
- api : cross
50 51
  inputs :
    {x : X, y : Y}
52 53 54 55 56
  attrs :
    axis : dim
  outputs :
    out : Out

57 58 59 60 61 62 63 64 65 66
- api : depthwise_conv2d
  backward : depthwise_conv2d_grad
  extra :
    attrs : [bool use_cudnn = false, bool fuse_relu_before_depthwise_conv = false, bool use_mkldnn = false,
             bool use_quantizer = false, str mkldnn_data_type = "float32", bool fuse_relu = false,
             str fuse_activation = "", float fuse_alpha = 0.0f, float fuse_beta = 0.0f, bool use_addto = false,
             bool fuse_residual_connection = false, float Scale_in = 1.0f, float Scale_out = 1.0f,
             float Scale_in_eltwise = 1.0f, 'float[] Scale_weights = {1.0f}', bool force_fp32_output = false,
             int workspace_size_MB = 512, bool exhaustive_search = false]

67 68 69 70 71 72 73 74
- api : diag
  op_name : diag_v2
  grad_op_name : diag_v2_grad
  inputs :
    x : X
  outputs :
    out : Out

75 76 77 78 79 80
- api : diagonal
  inputs :
    x : Input
  outputs :
    out : Out

81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
- api : digamma
  inputs :
    x : X
  outputs :
    out : Out

- api : dist
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

- api : dot
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

- api : erf
  inputs :
    x : X
  outputs :
    out : Out

105 106 107 108 109 110
- api : erfinv
  inputs :
    x : X
  outputs :
    out : Out

111 112 113 114 115 116
- api : lgamma
  inputs :
    x : X
  outputs :
    out : Out

117 118 119 120 121 122 123 124 125 126 127 128
- api : mv
  inputs :
    {x : X, vec : Vec}
  outputs :
    out : Out

- api : poisson
  inputs :
    x : X
  outputs :
    out : Out

129 130 131 132 133 134
- api : solve
  inputs :
    {x : X, y : Y}
  outputs :
    out : Out

135 136 137 138 139
- api : trace
  inputs :
    x : Input
  outputs :
    out : Out
140

141 142 143 144 145
- api : trunc
  inputs :
    x : X
  outputs :
    out : Out
F
Feiyu Chan 已提交
146 147 148 149 150 151 152 153 154 155 156 157

- api: fft_c2c
  inputs: {x: X}
  outputs: {out: Out}

- api: fft_c2r
  inputs: {x: X}
  outputs: {out: Out}

- api: fft_r2c
  inputs: {x: X}
  outputs: {out: Out}