pd_op.yaml 797 字节
Newer Older
1
- name: feed
K
kangguangli 已提交
2
  inputs: []
3
  attrs:
K
kangguangli 已提交
4
  - {typename: str, name: name}
H
hong 已提交
5
  - {typename: int, name: col}
6 7 8 9 10 11 12 13 14 15 16 17 18 19
  outputs:
  - {typename: Tensor, name: out, optional: false, intermediate: false}
  no_need_buffer: null
  data_transform: null
  inplace: null
  backward: null
- name: fetch
  inputs:
  - typename: Tensor
    name: x
    optional: false
    no_need_buffer: false
    data_transform: {}
  attrs:
K
kangguangli 已提交
20
  - {typename: str, name: name}
H
hong 已提交
21 22
  outputs:
  - {typename: Tensor, name: out, optional: false, intermediate: false}
23 24
  no_need_buffer: null
  data_transform: null
H
hong 已提交
25 26 27 28 29 30 31 32 33 34 35
  infer_meta:
    func: UnchangedInferMeta
    param: [x]
  kernel:
    func: [fetch]
    param: [x]
    backend: null
    layout: null
    data_type: null
    dispatch: {fetch: null}
    force_backend: null
36 37
  inplace: null
  backward: null