未验证 提交 7101af3f 编写于 作者: W Wangzheee 提交者: GitHub

Add the op def for batch_norm, conv2d_transpose (#33360)

上级 d19bceb6
type: "batch_norm"
def {
inputs {
name: "X"
}
inputs {
name: "Scale"
}
inputs {
name: "Bias"
}
inputs {
name: "Mean"
}
inputs {
name: "Variance"
}
outputs {
name: "Y"
}
attrs {
name: "epsilon"
type: FLOAT
}
}
extra {
inputs {
name: "MomentumTensor"
}
attrs {
name: "is_test"
type: BOOLEAN
}
attrs {
name: "momentum"
type: FLOAT
}
attrs {
name: "data_layout"
type: STRING
}
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs {
name: "fuse_with_relu"
type: BOOLEAN
}
attrs {
name: "use_global_stats"
type: BOOLEAN
}
attrs {
name: "trainable_statistics"
type: BOOLEAN
}
outputs {
name: "MeanOut"
}
outputs {
name: "VarianceOut"
}
outputs {
name: "SavedMean"
}
outputs {
name: "SavedVariance"
}
outputs {
name: "ReserveSpace"
}
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
}
type: "reduce_mean"
def {
inputs {
name: "Input"
}
inputs {
name: "Filter"
}
inputs {
name: "Bias"
}
outputs {
name: "Output"
}
attrs {
name: "output_padding"
type: INTS
}
attrs {
name: "output_size"
type: INTS
}
attrs {
name: "groups"
type: INT
}
attrs {
name: "dilations"
type: INTS
}
attrs {
name: "strides"
type: INTS
}
attrs {
name: "paddings"
type: INTS
}
attrs {
name: "padding_algorithm"
type: STRING
}
}
extra {
attrs {
name: "is_test"
type: BOOLEAN
}
attrs {
name: "use_cudnn"
type: BOOLEAN
}
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs {
name: "force_fp32_output"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs {
name: "fuse_relu"
type: BOOLEAN
}
attrs {
name: "fuse_activation"
type: STRING
}
attrs {
name: "fuse_alpha"
type: FLOAT
}
attrs {
name: "fuse_beta"
type: FLOAT
}
attrs {
name: "data_format"
type: STRING
}
attrs {
name: "workspace_size_MB"
type: INT
}
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册