api_compat.yaml 1.7 KB
Newer Older
1 2
- api : atan2
  inputs :
3
    {x : X1, y : X2}
4 5 6
  outputs :
    out : Out

7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
- 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
  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 = "", bool fuse_alpha = false, bool fuse_beta = false, 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]

34
- api : cross
35 36
  inputs :
    {x : X, y : Y}
37 38 39 40 41
  attrs :
    axis : dim
  outputs :
    out : Out

42 43 44 45 46 47 48 49
- api : diag
  op_name : diag_v2
  grad_op_name : diag_v2_grad
  inputs :
    x : X
  outputs :
    out : Out

50 51 52 53 54 55
- api : diagonal
  inputs :
    x : Input
  outputs :
    out : Out

56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
- 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

- api : mv
  inputs :
    {x : X, vec : Vec}
  outputs :
    out : Out

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

92 93 94 95 96
- api : trace
  inputs :
    x : Input
  outputs :
    out : Out
97

98 99 100 101 102
- api : trunc
  inputs :
    x : X
  outputs :
    out : Out