Home
last modified time | relevance | path

Searched refs:TensorDict (Results 1 – 1 of 1) sorted by relevance

/test/mlts/benchmark/tools/
Dtensor_utils.py194 class TensorDict(dict): class
296 self.tensor_dict = TensorDict(self.model_dir)