Searched defs:tensor_or_dict (Results 1 – 1 of 1) sorted by relevance
250 def _extract_key_names(tensor_or_dict): argument3655 def nest_pad(tensor_or_dict): argument3668 def slice_tensor_or_dict(tensor_or_dict, signals): argument