tensorflow / runtime

A performant and modular runtime for TensorFlow
Apache License 2.0
756 stars 123 forks source link

reshape operation in TFRT #106

Open JasonMaojinsong opened 2 years ago

JasonMaojinsong commented 2 years ago

Hi I want to reshape the input feature data of mnist.mlir ("integrationtest/mnist/test_data/mnist_tensors.btf") which has [100, 784] into a shape such as [100,28,28,1], and follow that with a conv operation, but I can't find any reshape example in mlir test, could anyone give me some suggestions? does current TFRT support this operation?