tvm-dev mailing list archives

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


Document is at https://www.tensorflow.org/api_docs/python/tf/TensorArray

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](https://discuss.tvm.ai/t/how-to-support-tf-tensorarray/1983/1) to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click here](https://discuss.tvm.ai/email/unsubscribe/f031cc59c9f0ebff847a27b33edf68e89f2ffd1c5b88b1c70a778e48ba8409b8).

Tianqi Chen, UW, Seattle, WA, 98105, United States
http://tracking.discuss.tvm.ai/tracking/unsubscribe?msgid=EA5YreDKfw2Gr1UVP_i_Zg2
Mime
  • Unnamed multipart/alternative (inline, None, 0 bytes)
View raw message