From 2561c27de32dfe93dac68310a8d912b43a784c5c Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 6 Jan 2020 11:22:17 -0800 Subject: [PATCH] Fixed typo. Removed outdated comment. PiperOrigin-RevId: 288338957 Change-Id: I3f7d7d7cc4401643e143e3ea01a7513569062329 --- tensorflow/lite/delegates/gpu/cl/kernels/util.h | 6 +++--- tensorflow/lite/delegates/gpu/cl/tensor_type.cc | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tensorflow/lite/delegates/gpu/cl/kernels/util.h b/tensorflow/lite/delegates/gpu/cl/kernels/util.h index 3d3618c1995..c7ee5333fa6 100644 --- a/tensorflow/lite/delegates/gpu/cl/kernels/util.h +++ b/tensorflow/lite/delegates/gpu/cl/kernels/util.h @@ -97,7 +97,6 @@ class TensorCodeGenerator { std::string Write3D(const std::string& var_name, const std::string& x, 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, const std::string& y, const std::string& z, const std::string& b) const; @@ -133,8 +132,9 @@ std::string GetTensorDeclaration(AccessType access, const std::string& tensor_name, const TensorDescriptor& descriptor); -// Calculates correct X coordinate when stride != 1 and batch != 1 for -// DHWBC4, HDWBC4, HWBC layouts +// Calculates correct X coordinate when stride != 1 and batch != 1 for 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, const std::string& batch_size, const std::string& stride_x, diff --git a/tensorflow/lite/delegates/gpu/cl/tensor_type.cc b/tensorflow/lite/delegates/gpu/cl/tensor_type.cc index 8ca73c2dec7..900bf6e620d 100644 --- a/tensorflow/lite/delegates/gpu/cl/tensor_type.cc +++ b/tensorflow/lite/delegates/gpu/cl/tensor_type.cc @@ -30,7 +30,7 @@ std::string ToString(TensorStorageType type) { case TensorStorageType::TEXTURE_2D: return "TensorStorageType::TEXTURE_2D"; case TensorStorageType::TEXTURE_3D: - return "TensorStorageType::TEXTURE_2D"; + return "TensorStorageType::TEXTURE_3D"; case TensorStorageType::SINGLE_TEXTURE_2D: return "TensorStorageType::SINGLE_TEXTURE_2D"; case TensorStorageType::IMAGE_BUFFER: