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 } }