Searched refs:map_tensor_sizes (Results 1 – 1 of 1) sorted by relevance
3380 def map_tensor_sizes(sizes): function3382 return [map_tensor_sizes(s) for s in sizes]3388 self._arg_cache[name] = map_tensor_sizes(self._extra_kwargs[size_name])