| op { |
| name: "MaxPoolGradWithArgmax" |
| input_arg { |
| name: "input" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "grad" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "argmax" |
| type_attr: "Targmax" |
| } |
| output_arg { |
| name: "output" |
| type_attr: "T" |
| } |
| attr { |
| name: "ksize" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "strides" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "padding" |
| type: "string" |
| allowed_values { |
| list { |
| s: "SAME" |
| s: "VALID" |
| } |
| } |
| } |
| attr { |
| name: "Targmax" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_INT32 |
| type: DT_INT64 |
| } |
| } |
| } |
| attr { |
| name: "T" |
| type: "type" |
| default_value { |
| type: DT_FLOAT |
| } |
| allowed_values { |
| list { |
| type: DT_FLOAT |
| type: DT_HALF |
| } |
| } |
| } |
| } |
| op { |
| name: "MaxPoolGradWithArgmax" |
| input_arg { |
| name: "input" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "grad" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "argmax" |
| type_attr: "Targmax" |
| } |
| output_arg { |
| name: "output" |
| type_attr: "T" |
| } |
| attr { |
| name: "ksize" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "strides" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "padding" |
| type: "string" |
| allowed_values { |
| list { |
| s: "SAME" |
| s: "VALID" |
| } |
| } |
| } |
| attr { |
| name: "Targmax" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_INT32 |
| type: DT_INT64 |
| } |
| } |
| } |
| attr { |
| name: "T" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_FLOAT |
| type: DT_DOUBLE |
| type: DT_INT32 |
| type: DT_INT64 |
| type: DT_UINT8 |
| type: DT_INT16 |
| type: DT_INT8 |
| type: DT_UINT16 |
| type: DT_HALF |
| } |
| } |
| } |
| } |
| op { |
| name: "MaxPoolGradWithArgmax" |
| input_arg { |
| name: "input" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "grad" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "argmax" |
| type_attr: "Targmax" |
| } |
| output_arg { |
| name: "output" |
| type_attr: "T" |
| } |
| attr { |
| name: "ksize" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "strides" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "padding" |
| type: "string" |
| allowed_values { |
| list { |
| s: "SAME" |
| s: "VALID" |
| } |
| } |
| } |
| attr { |
| name: "Targmax" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_INT32 |
| type: DT_INT64 |
| } |
| } |
| } |
| attr { |
| name: "T" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_FLOAT |
| type: DT_DOUBLE |
| type: DT_INT32 |
| type: DT_INT64 |
| type: DT_UINT8 |
| type: DT_INT16 |
| type: DT_INT8 |
| type: DT_UINT16 |
| type: DT_HALF |
| type: DT_UINT32 |
| type: DT_UINT64 |
| } |
| } |
| } |
| } |
| op { |
| name: "MaxPoolGradWithArgmax" |
| input_arg { |
| name: "input" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "grad" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "argmax" |
| type_attr: "Targmax" |
| } |
| output_arg { |
| name: "output" |
| type_attr: "T" |
| } |
| attr { |
| name: "ksize" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "strides" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "padding" |
| type: "string" |
| allowed_values { |
| list { |
| s: "SAME" |
| s: "VALID" |
| } |
| } |
| } |
| attr { |
| name: "Targmax" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_INT32 |
| type: DT_INT64 |
| } |
| } |
| } |
| attr { |
| name: "T" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_FLOAT |
| type: DT_DOUBLE |
| type: DT_INT32 |
| type: DT_INT64 |
| type: DT_UINT8 |
| type: DT_INT16 |
| type: DT_INT8 |
| type: DT_UINT16 |
| type: DT_HALF |
| type: DT_UINT32 |
| type: DT_UINT64 |
| type: DT_BFLOAT16 |
| } |
| } |
| } |
| } |
| op { |
| name: "MaxPoolGradWithArgmax" |
| input_arg { |
| name: "input" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "grad" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "argmax" |
| type_attr: "Targmax" |
| } |
| output_arg { |
| name: "output" |
| type_attr: "T" |
| } |
| attr { |
| name: "ksize" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "strides" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "padding" |
| type: "string" |
| allowed_values { |
| list { |
| s: "SAME" |
| s: "VALID" |
| } |
| } |
| } |
| attr { |
| name: "Targmax" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_INT32 |
| type: DT_INT64 |
| } |
| } |
| } |
| attr { |
| name: "T" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_FLOAT |
| type: DT_DOUBLE |
| type: DT_INT32 |
| type: DT_UINT8 |
| type: DT_INT16 |
| type: DT_INT8 |
| type: DT_INT64 |
| type: DT_BFLOAT16 |
| type: DT_UINT16 |
| type: DT_HALF |
| type: DT_UINT32 |
| type: DT_UINT64 |
| } |
| } |
| } |
| } |
| op { |
| name: "MaxPoolGradWithArgmax" |
| input_arg { |
| name: "input" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "grad" |
| type_attr: "T" |
| } |
| input_arg { |
| name: "argmax" |
| type_attr: "Targmax" |
| } |
| output_arg { |
| name: "output" |
| type_attr: "T" |
| } |
| attr { |
| name: "ksize" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "strides" |
| type: "list(int)" |
| has_minimum: true |
| minimum: 4 |
| } |
| attr { |
| name: "padding" |
| type: "string" |
| allowed_values { |
| list { |
| s: "SAME" |
| s: "VALID" |
| } |
| } |
| } |
| attr { |
| name: "include_batch_in_index" |
| type: "bool" |
| default_value { |
| b: false |
| } |
| } |
| attr { |
| name: "Targmax" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_INT32 |
| type: DT_INT64 |
| } |
| } |
| } |
| attr { |
| name: "T" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_FLOAT |
| type: DT_DOUBLE |
| type: DT_INT32 |
| type: DT_UINT8 |
| type: DT_INT16 |
| type: DT_INT8 |
| type: DT_INT64 |
| type: DT_BFLOAT16 |
| type: DT_UINT16 |
| type: DT_HALF |
| type: DT_UINT32 |
| type: DT_UINT64 |
| } |
| } |
| } |
| } |