Searched defs:gather_leaf_tensors (Results 1 – 2 of 2) sorted by relevance
403 def gather_leaf_tensors(args, kwargs): function
328 def gather_leaf_tensors(args, kwargs): function