Searched refs:node_calls (Results 1 – 2 of 2) sorted by relevance
481 compiler_call.node_calls in _compiled_autograd_impl()497 NodeCall& call = compiler_call.node_calls.lookup(fn); in _compiled_autograd_impl()533 compiler_call.node_calls.lookup(edge.function).needed = false; in _compiled_autograd_impl()
205 NodeCalls node_calls; member383 collect(_compiler.node_calls.lookup(t)); in collect()393 collect_size(_compiler.node_calls.lookup(t.function).id); in collect()