56 lines
686 B
Protocol Buffer Text Format
56 lines
686 B
Protocol Buffer Text Format
type: "softmax"
|
|
def {
|
|
inputs {
|
|
name: "X"
|
|
}
|
|
outputs {
|
|
name: "Out"
|
|
}
|
|
attrs {
|
|
name: "axis"
|
|
type: INT
|
|
}
|
|
}
|
|
extra {
|
|
attrs {
|
|
name: "data_format"
|
|
type: STRING
|
|
}
|
|
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
|
|
}
|
|
attrs {
|
|
name: "is_test"
|
|
type: BOOLEAN
|
|
}
|
|
attrs {
|
|
name: "mkldnn_data_type"
|
|
type: STRING
|
|
}
|
|
attrs {
|
|
name: "use_cudnn"
|
|
type: BOOLEAN
|
|
}
|
|
attrs {
|
|
name: "use_mkldnn"
|
|
type: BOOLEAN
|
|
}
|
|
}
|