| op { | |
| name: "CompareAndBitpack" | |
| input_arg { | |
| name: "input" | |
| type_attr: "T" | |
| } | |
| input_arg { | |
| name: "threshold" | |
| type_attr: "T" | |
| } | |
| output_arg { | |
| name: "output" | |
| type: DT_UINT8 | |
| } | |
| attr { | |
| name: "T" | |
| type: "type" | |
| allowed_values { | |
| list { | |
| type: DT_BOOL | |
| type: DT_HALF | |
| type: DT_FLOAT | |
| type: DT_DOUBLE | |
| type: DT_INT8 | |
| type: DT_INT16 | |
| type: DT_INT32 | |
| type: DT_INT64 | |
| } | |
| } | |
| } | |
| } |