How about this for mixed TIR and Relay:
class MixedModule:
@relay.script
def relay_func(x: ty.Tensor):
return relay.call_tir_dest_passing(tir_func, x)
@tir.script
def tir_func(x: ty.handle)
...
How about this for mixed TIR and Relay:
class MixedModule:
@relay.script
def relay_func(x: ty.Tensor):
return relay.call_tir_dest_passing(tir_func, x)
@tir.script
def tir_func(x: ty.handle)
...