tvm-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From Lixiaoquan via TVM Discuss <>
Subject [TVM Discuss] [Development] How to support tf.TensorArray()
Date Fri, 22 Mar 2019 08:08:33 GMT

Document is at

This code snippet generates several tensorflow ops:
[TensorArrayV3, TensorArrayGatherV3, TensorArraySizeV3, TensorArrayWriteV3]

tensor_array = tf.TensorArray(dtype=tf.float32, size=1, dynamic_size=True)

tensor_array = tensor_array.write(tf.constant(0), [2.0, 3.0])
tensor_array = tensor_array.write(tf.constant(1), [4.0, 5.0])

out = tensor_array.stack()

One possibility to support these is to enhance relay.Tuple. relay.Tuple is dynamic, but it
lacks ability to update items in it and it's size can't be get in runtime. 

Any suggestion about this issue?

[Visit Topic]( to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click here](

Tianqi Chen, UW, Seattle, WA, 98105, United States
  • Unnamed multipart/alternative (inline, None, 0 bytes)
View raw message