Searched refs:GetFastestStorageType (Results 1 – 4 of 4) sorted by relevance
78 TensorStorageType GetFastestStorageType(const GpuInfo& gpu_info);
187 TensorStorageType GetFastestStorageType(const GpuInfo& gpu_info) { in GetFastestStorageType() function
631 preferred_storage_types = {GetFastestStorageType(env.device().GetInfo()), in GetStorageTypeFromOptions()
46 create_info.storage_type = GetFastestStorageType(env.device().GetInfo()); in RunModelSample()