experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added handling of Intel in choosing best storage types.

Browse Source 
PiperOrigin-RevId: 316550810
Change-Id: I7e81cd0df1522be4b705df57a5658397328b5a18
This commit is contained in:
Raman Sarokin 2020-06-15 14:54:50 -07:00 committed by TensorFlower Gardener
parent 92bd07fee5
commit 176ab11d0a
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/lite/delegates/gpu/cl/environment.cc
View File

@ -242,6 +242,8 @@ TensorStorageType GetFastestStorageType(const CLDevice& gpu) {
} else if (gpu.IsAMD()) { } else if (gpu.IsAMD()) {
return gpu.SupportsImageBuffer() ? TensorStorageType::IMAGE_BUFFER return gpu.SupportsImageBuffer() ? TensorStorageType::IMAGE_BUFFER
: TensorStorageType::BUFFER; : TensorStorageType::BUFFER;
} else if (gpu.IsIntel()) {
return TensorStorageType::BUFFER;
} }
return TensorStorageType::BUFFER; return TensorStorageType::BUFFER;
} }
@ -264,6 +266,8 @@ TensorStorageType GetStorageTypeWithMinimalMemoryConsumption(
} else if (gpu.IsAMD()) { } else if (gpu.IsAMD()) {
return gpu.SupportsImageBuffer() ? TensorStorageType::IMAGE_BUFFER return gpu.SupportsImageBuffer() ? TensorStorageType::IMAGE_BUFFER
: TensorStorageType::BUFFER; : TensorStorageType::BUFFER;
} else if (gpu.IsIntel()) {
return TensorStorageType::BUFFER;
} }
return TensorStorageType::BUFFER; return TensorStorageType::BUFFER;
} }