Tensorflow: создайте tf.NodeDef() и установите атрибуты

Я пытаюсь создать новый узел и установить его атрибуты.

Например, печатая один из узлов графа, я вижу, что его атрибуты:

attr {
  key: "T"
  value {
    type: DT_FLOAT
  }
}

Я могу создать узел, например:

node = tf.NodeDef(name='MyConstTensor', op='Const',
                   attr = {'value': tf.AttrValue(tensor=tensor_proto),
                         'dtype': tf.AttrValue(type=dt)})

но как добавить атрибут key: "T"? то есть что должно быть внутри tf.AttrValue в этом случае?

Глядя на attr_value.proto, я пробовал:

node = tf.NodeDef()
node.name = 'MySub'
node.op = 'Sub'
node.input.extend(['MyConstTensor', 'conv2'])
node.attr["key"].s = 'T' # TypeError: 'T' has type str, but expected one of: bytes

Обновлено:

Я понял, что в Tensorflow это должно быть написано так:

node.attr["T"].type = b'float32'

Но это дает ошибку:

TypeError: b'float32' has type bytes, but expected one of: int, long

И я не уверен, какое значение int соответствует float32.

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/framework/attr_value.proto#L23

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/framework/attr_value.proto#L35