Searched refs:normalize_to_args (Results 1 – 1 of 1) sorted by relevance
1461 def normalize_to_args(args, kwargs): member in TemplatedAttentionHigherOrderVariable1538 query, key, value, score_mod = self.normalize_to_args(args, kwargs)