| 185 | } |
| 186 | |
| 187 | std::vector<TensorStorageType> Environment::GetSupportedStorages() const { |
| 188 | std::vector<TensorStorageType> storage_types = {TensorStorageType::TEXTURE_2D, |
| 189 | TensorStorageType::BUFFER}; |
| 190 | if (device_.SupportsTextureArray()) { |
| 191 | storage_types.push_back(TensorStorageType::TEXTURE_ARRAY); |
| 192 | } |
| 193 | return storage_types; |
| 194 | } |
| 195 | |
| 196 | TensorStorageType GetOptimalStorageType(const CLDevice& gpu) { |
| 197 | TensorStorageType storage_type; |