How to customize node names?

Troubleshooting
1

Hi everybody,

I’m looking into how to use custom node names into TVM models.

For example I’d like to add a prefix or a suffix, e.g. fused_nn_contrib_conv2d_NCHWc_add_nn_relu_9 becomes prefix_fused_nn_contrib_conv2d_NCHWc_add_nn_relu_9_postfix.

Can anybody point me to the right piece of code that I can modify to make this possible?

Many thanks.

2

perhaps try here: https://github.com/apache/tvm/blob/main/src/relay/backend/compile_engine.cc#L130

1 Like