Call nn.Graph subclass instance to run your customized graph.
Call your customized graph after the instantiation:
g = CustomGraph() out_tensors = g(input_tensors)
The inputs of
__call__method must match the inputs of
build()method. And the
__call__method will return outputs matching the outputs of
The first call takes longer than later calls, because nn.Graph will do the computaion graph generation and optimization at the first call.
Donot override this function.