static_ops.yaml 490 字节
Newer Older
1 2 3 4 5 6 7 8 9
- op : fc_xpu
  args : (Tensor x, Tensor w, Tensor w_max, Tensor bias, int in_num_col_dims, bool transpose_x, float alpha, float beta, int act_type, float act_alpha)
  output : Tensor
  infer_meta :
    func : FcXPUInferMeta
  kernel :
    func : fc_xpu
  optional : bias

10 11 12 13 14 15 16
- op : share_buffer
  args : (Tensor[] x, bool[] share_dims_and_dtype={})
  output : Tensor[](out){x.size()}, Tensor[](xout){x.size()}
  infer_meta :
    func : ShareBufferInferMeta
  kernel :
    func : share_buffer