Assume you have B = topi.nn.relu(A)
, I want to set the name of B like how we name the result with B = tvm.compute(name='customize_output_0', ..)
.
Is it possible to do this for topi ops?
Assume you have B = topi.nn.relu(A)
, I want to set the name of B like how we name the result with B = tvm.compute(name='customize_output_0', ..)
.
Is it possible to do this for topi ops?