api_compat.yaml 928 字节
Newer Older
1 2 3 4 5 6 7
- api : atan2
  inputs :
    x : X1
    y : X2
  outputs :
    out : Out

8 9 10 11 12 13 14 15 16 17 18 19 20
- api : cross
  inputs : {x : X, y : Y}
  attrs :
    axis : dim
  outputs :
    out : Out

- api : diagonal
  inputs :
    x : Input
  outputs :
    out : Out

21 22 23 24 25
- api : trace
  inputs :
    x : Input
  outputs :
    out : Out
26 27 28 29 30 31 32 33 34

- 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]