| op { |
| name: "AccumulatorTakeGradient" |
| input_arg { |
| name: "handle" |
| type: DT_STRING |
| is_ref: true |
| } |
| input_arg { |
| name: "num_required" |
| type: DT_INT32 |
| } |
| output_arg { |
| name: "average" |
| type_attr: "dtype" |
| } |
| attr { |
| name: "dtype" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_FLOAT |
| type: DT_DOUBLE |
| type: DT_INT64 |
| type: DT_INT32 |
| type: DT_UINT8 |
| type: DT_UINT16 |
| type: DT_INT16 |
| type: DT_INT8 |
| type: DT_COMPLEX64 |
| type: DT_COMPLEX128 |
| type: DT_QINT8 |
| type: DT_QUINT8 |
| type: DT_QINT32 |
| type: DT_HALF |
| } |
| } |
| } |
| } |
| op { |
| name: "AccumulatorTakeGradient" |
| input_arg { |
| name: "handle" |
| type: DT_STRING |
| is_ref: true |
| } |
| input_arg { |
| name: "num_required" |
| type: DT_INT32 |
| } |
| output_arg { |
| name: "average" |
| type_attr: "dtype" |
| } |
| attr { |
| name: "dtype" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_FLOAT |
| type: DT_DOUBLE |
| type: DT_INT64 |
| type: DT_INT32 |
| type: DT_UINT8 |
| type: DT_UINT16 |
| type: DT_INT16 |
| type: DT_INT8 |
| type: DT_COMPLEX64 |
| type: DT_COMPLEX128 |
| type: DT_QINT8 |
| type: DT_QUINT8 |
| type: DT_QINT32 |
| type: DT_HALF |
| type: DT_UINT32 |
| type: DT_UINT64 |
| } |
| } |
| } |
| } |
| op { |
| name: "AccumulatorTakeGradient" |
| input_arg { |
| name: "handle" |
| type: DT_STRING |
| is_ref: true |
| } |
| input_arg { |
| name: "num_required" |
| type: DT_INT32 |
| } |
| output_arg { |
| name: "average" |
| type_attr: "dtype" |
| } |
| attr { |
| name: "dtype" |
| type: "type" |
| allowed_values { |
| list { |
| type: DT_FLOAT |
| type: DT_DOUBLE |
| type: DT_INT64 |
| type: DT_INT32 |
| type: DT_UINT8 |
| type: DT_UINT16 |
| type: DT_INT16 |
| type: DT_INT8 |
| type: DT_COMPLEX64 |
| type: DT_COMPLEX128 |
| type: DT_QINT8 |
| type: DT_QUINT8 |
| type: DT_QINT32 |
| type: DT_HALF |
| type: DT_UINT32 |
| type: DT_UINT64 |
| type: DT_BFLOAT16 |
| } |
| } |
| } |
| } |
| op { |
| name: "AccumulatorTakeGradient" |
| input_arg { |
| name: "handle" |
| type: DT_STRING |
| is_ref: true |
| } |
| input_arg { |
| name: "num_required" |
| type: DT_INT32 |
| } |
| output_arg { |
| name: "average" |
| type_attr: "dtype" |
| } |
| 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 |
| } |
| } |
| } |
| } |