[TVM Discuss] [Questions] Autotvm.task_extract_from_program in TFLite

[Abelardo López-Lagunas via TVM Discuss]

Thanks!, I forgot to add that I did the `ConvertLayout` pass before to 
transform into NCHW:
```
 mod, params =\
        relay.frontend.from_tflite(tflite_model,
                                   shape_dict={input_name: dshape},
                                   dtype_dict={input_name: input_type})
    if layout == "NCHW":
        logging.warning("Applying NHWC to NCHW transformation")
        # This is the transformation pass
        seq = tvm.transform.Sequential([
            relay.transform.RemoveUnusedFunctions(),
            relay.transform.ConvertLayout(layout)
        ])
        mod = seq(mod)
        logging.warning("NHWC to NCHW transformation done")
```





---
[Visit 
Topic](https://discuss.tvm.ai/t/autotvm-task-extract-from-program-in-tflite/6578/3)
 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/df8f2fd0536a5aef89c90a0490c59ee75f4d850a4b46858d27987f70e45ada77).