Searched refs:GetFastestStorageType (Results 1 – 8 of 8) sorted by relevance
48 create_info.storage_type = GetFastestStorageType(env.device().GetInfo()); in RunPredefinedLayoutSample()92 create_info.storage_type = GetFastestStorageType(env.device().GetInfo()); in RunExternalImmutableSample()144 create_info.storage_type = GetFastestStorageType(env.device().GetInfo()); in RunSerializedTest()289 create_info.storage_type = GetFastestStorageType(env.device().GetInfo()); in RunModelSample()
78 TensorStorageType GetFastestStorageType(const GpuInfo& gpu_info);
188 TensorStorageType GetFastestStorageType(const GpuInfo& gpu_info) { in GetFastestStorageType() function
643 preferred_storage_types = {GetFastestStorageType(env.device().GetInfo()), in GetStorageTypeFromOptions()
122 TensorStorageType GetFastestStorageType(const GpuInfo& gpu_info);
411 TensorStorageType GetFastestStorageType(const GpuInfo& gpu_info) { in GetFastestStorageType() function
138 create_info.storage_type = GetFastestStorageType(gpu_info);205 create_info.storage_type = GetFastestStorageType(gpu_info);271 create_info.storage_type = GetFastestStorageType(gpu_info);
361 create_info.storage_type = GetFastestStorageType(gpu_info);