fused_backward.yaml 911 字节
Newer Older
1
# This file is designed for fusion C++ backward operators, which manages the
Z
zyfncg 已提交
2
# generated code for static mode and dynamic mode (when `support_dygraph_mode` is true).
Z
zyfncg 已提交
3
# "support_dygraph_mode" is an extra configuration item in this file,
Z
zyfncg 已提交
4 5
# if one operator have "support_dygraph_mode : true", it supports dygraph mode,
# otherwise the operator only could be used in static mode.
6 7

- backward_op : fused_dropout_add_grad
8
  forward : fused_dropout_add (Tensor x, Tensor y, Tensor seed_tensor, Scalar p, bool is_test, str mode, int seed, bool fix_seed) -> Tensor(out), Tensor(seed_offset)
9 10 11
  args : (Tensor seed_offset, Tensor out_grad, Scalar p, bool is_test, str mode, bool fix_seed)
  output : Tensor(x_grad), Tensor(y_grad)
  infer_meta :
12 13
    func : FusedDropoutAddGradInferMeta
    param : [seed_offset, out_grad]
14 15
  kernel :
    func : fused_dropout_add_grad
16
    data_type : out_grad
17
  support_dygraph_mode : true