Searched refs:concat_inner_tensors_from_subclasses (Results 1 – 1 of 1) sorted by relevance
115 def concat_inner_tensors_from_subclasses(xs): function129 unwrapped_args_fw = concat_inner_tensors_from_subclasses(wrapped_args[0])130 unwrapped_args_tangents = concat_inner_tensors_from_subclasses(wrapped_args[1])134 unwrapped_args_fw = concat_inner_tensors_from_subclasses(wrapped_args)