Searched refs:maybe_get_next_module (Results 1 – 3 of 3) sorted by relevance
19 maybe_get_next_module,
23 maybe_get_next_module,421 maybe_relu_node = maybe_get_next_module(node, modules, nn.ReLU)423 maybe_relu_node = maybe_get_next_module(430 maybe_get_next_module(node, modules, ObserverBase)432 else maybe_get_next_module(maybe_relu_node, modules, ObserverBase)437 maybe_eq_obs_node = maybe_get_next_module(
419 def maybe_get_next_module( function