| op { | |
| name: "ResourceConditionalAccumulator" | |
| output_arg { | |
| name: "handle" | |
| type: DT_RESOURCE | |
| } | |
| attr { | |
| name: "dtype" | |
| 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_COMPLEX64 | |
| type: DT_INT64 | |
| type: DT_QINT8 | |
| type: DT_QUINT8 | |
| type: DT_QINT32 | |
| type: DT_BFLOAT16 | |
| type: DT_UINT16 | |
| type: DT_COMPLEX128 | |
| type: DT_HALF | |
| type: DT_UINT32 | |
| type: DT_UINT64 | |
| } | |
| } | |
| } | |
| attr { | |
| name: "shape" | |
| type: "shape" | |
| } | |
| attr { | |
| name: "container" | |
| type: "string" | |
| default_value { | |
| s: "" | |
| } | |
| } | |
| attr { | |
| name: "shared_name" | |
| type: "string" | |
| default_value { | |
| s: "" | |
| } | |
| } | |
| attr { | |
| name: "reduction_type" | |
| type: "string" | |
| default_value { | |
| s: "MEAN" | |
| } | |
| allowed_values { | |
| list { | |
| s: "MEAN" | |
| s: "SUM" | |
| } | |
| } | |
| } | |
| is_stateful: true | |
| } |