Searched refs:_alloc_tensor (Results 1 – 1 of 1) sorted by relevance
99 def _alloc_tensor( function299 state_dict[key] = _alloc_tensor(304 _alloc_tensor(value.properties, value.size, dp_pg_device_type),330 tensor=_alloc_tensor(