op { | |
name: "ApplyAdam" | |
input_arg { | |
name: "var" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "m" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "v" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "beta1_power" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta2_power" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "lr" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta1" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta2" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "epsilon" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "grad" | |
type_attr: "T" | |
} | |
output_arg { | |
name: "out" | |
type_attr: "T" | |
is_ref: true | |
} | |
attr { | |
name: "T" | |
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 | |
} | |
} | |
} | |
attr { | |
name: "use_locking" | |
type: "bool" | |
default_value { | |
b: false | |
} | |
} | |
} | |
op { | |
name: "ApplyAdam" | |
input_arg { | |
name: "var" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "m" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "v" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "beta1_power" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta2_power" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "lr" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta1" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta2" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "epsilon" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "grad" | |
type_attr: "T" | |
} | |
output_arg { | |
name: "out" | |
type_attr: "T" | |
is_ref: true | |
} | |
attr { | |
name: "T" | |
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 | |
} | |
} | |
} | |
attr { | |
name: "use_locking" | |
type: "bool" | |
default_value { | |
b: false | |
} | |
} | |
attr { | |
name: "use_nesterov" | |
type: "bool" | |
default_value { | |
b: false | |
} | |
} | |
} | |
op { | |
name: "ApplyAdam" | |
input_arg { | |
name: "var" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "m" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "v" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "beta1_power" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta2_power" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "lr" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta1" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta2" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "epsilon" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "grad" | |
type_attr: "T" | |
} | |
output_arg { | |
name: "out" | |
type_attr: "T" | |
is_ref: true | |
} | |
attr { | |
name: "T" | |
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 | |
} | |
} | |
} | |
attr { | |
name: "use_locking" | |
type: "bool" | |
default_value { | |
b: false | |
} | |
} | |
attr { | |
name: "use_nesterov" | |
type: "bool" | |
default_value { | |
b: false | |
} | |
} | |
} | |
op { | |
name: "ApplyAdam" | |
input_arg { | |
name: "var" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "m" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "v" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "beta1_power" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta2_power" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "lr" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta1" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta2" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "epsilon" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "grad" | |
type_attr: "T" | |
} | |
output_arg { | |
name: "out" | |
type_attr: "T" | |
is_ref: true | |
} | |
attr { | |
name: "T" | |
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 | |
} | |
} | |
} | |
attr { | |
name: "use_locking" | |
type: "bool" | |
default_value { | |
b: false | |
} | |
} | |
attr { | |
name: "use_nesterov" | |
type: "bool" | |
default_value { | |
b: false | |
} | |
} | |
} | |
op { | |
name: "ApplyAdam" | |
input_arg { | |
name: "var" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "m" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "v" | |
type_attr: "T" | |
is_ref: true | |
} | |
input_arg { | |
name: "beta1_power" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta2_power" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "lr" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta1" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "beta2" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "epsilon" | |
type_attr: "T" | |
} | |
input_arg { | |
name: "grad" | |
type_attr: "T" | |
} | |
output_arg { | |
name: "out" | |
type_attr: "T" | |
is_ref: true | |
} | |
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_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: "use_locking" | |
type: "bool" | |
default_value { | |
b: false | |
} | |
} | |
attr { | |
name: "use_nesterov" | |
type: "bool" | |
default_value { | |
b: false | |
} | |
} | |
} |