similar to introduced TensorRT in Tensorflow. I need to transform TF Dialect operations to TensorRT Dialect operations. I want to use DRR to achieve it. I found the ‘patternrewriter’ generated by DRR contains two steps: match and rewrite. The two steps is in the one function.
the ‘match’ step used to judge whether a TF Dialect operation can be converted to be a TensorRT Dialect operation.
the ‘rewrite’ step convert the TF Dialect operation to the TensorRT Dialect operation.
In fact, we only convert the TF Dialect operation to the TensorRT Dialect operation when the subgraph size is more than a specified number。 when we pick up the tensorrt subgraph, we need the “match step” result. So, I want to know is there a way to only generate the “match” step code?
best wishes to you!