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

Added missed cases for TEXTURE_3D.

Browse Source 
PiperOrigin-RevId: 288533435
Change-Id: I1f0b98877721446837f52fdca5f98326079b6b65
This commit is contained in:
A. Unique TensorFlower 2020-01-07 11:11:21 -08:00 committed by TensorFlower Gardener
parent febe171d32
commit e37290d8df
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/delegates/gpu/cl/tensor_type_util.cc
View File

@ -27,6 +27,7 @@ ObjectType ToObjectType(TensorStorageType type) {
case TensorStorageType::SINGLE_TEXTURE_2D:
case TensorStorageType::TEXTURE_2D:
case TensorStorageType::TEXTURE_ARRAY:
case TensorStorageType::TEXTURE_3D:
return ObjectType::OPENCL_TEXTURE;
default:
return ObjectType::UNKNOWN;
@ -45,6 +46,8 @@ DataLayout ToDataLayout(TensorStorageType type) {
return DataLayout::HDWC4;
case TensorStorageType::TEXTURE_ARRAY:
return DataLayout::DHWC4;
case TensorStorageType::TEXTURE_3D:
return DataLayout::DHWC4;
default:
return DataLayout::UNKNOWN;
}