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

Fixed typo. 

Removed outdated comment.

PiperOrigin-RevId: 288338957
Change-Id: I3f7d7d7cc4401643e143e3ea01a7513569062329
This commit is contained in:
A. Unique TensorFlower 2020-01-06 11:22:17 -08:00 committed by TensorFlower Gardener
parent d6e63d2151
commit 2561c27de3
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/delegates/gpu/cl/kernels/util.h
View File

@ -97,7 +97,6 @@ class TensorCodeGenerator {
std::string Write3D(const std::string& var_name, const std::string& x, std::string Write3D(const std::string& var_name, const std::string& x,
const std::string& y, const std::string& z) const; const std::string& y, const std::string& z) const;
// Write4D supports BUFFER and IMAGE_BUFFER storage types.
std::string Write4D(const std::string& var_name, const std::string& x, std::string Write4D(const std::string& var_name, const std::string& x,
const std::string& y, const std::string& z, const std::string& y, const std::string& z,
const std::string& b) const; const std::string& b) const;
@ -133,8 +132,9 @@ std::string GetTensorDeclaration(AccessType access,
const std::string& tensor_name, const std::string& tensor_name,
const TensorDescriptor& descriptor); const TensorDescriptor& descriptor);
// Calculates correct X coordinate when stride != 1 and batch != 1 for // Calculates correct X coordinate when stride != 1 and batch != 1 for layouts
// DHWBC4, HDWBC4, HWBC layouts // with B after W (for example HWBC4) and WB stored in one axis of GPU
// resources.
std::string GetXStrideCorrected(const std::string& src_x, std::string GetXStrideCorrected(const std::string& src_x,
const std::string& batch_size, const std::string& batch_size,
const std::string& stride_x, const std::string& stride_x,

2
tensorflow/lite/delegates/gpu/cl/tensor_type.cc
View File

@ -30,7 +30,7 @@ std::string ToString(TensorStorageType type) {
case TensorStorageType::TEXTURE_2D: case TensorStorageType::TEXTURE_2D:
return "TensorStorageType::TEXTURE_2D"; return "TensorStorageType::TEXTURE_2D";
case TensorStorageType::TEXTURE_3D: case TensorStorageType::TEXTURE_3D:
return "TensorStorageType::TEXTURE_2D"; return "TensorStorageType::TEXTURE_3D";
case TensorStorageType::SINGLE_TEXTURE_2D: case TensorStorageType::SINGLE_TEXTURE_2D:
return "TensorStorageType::SINGLE_TEXTURE_2D"; return "TensorStorageType::SINGLE_TEXTURE_2D";
case TensorStorageType::IMAGE_BUFFER: case TensorStorageType::IMAGE_BUFFER: