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

Introduce Vulkan API with integration tests.

Browse Source 
PiperOrigin-RevId: 313262552
Change-Id: I7d56bba03b7752938bd5f0cf5b08315941118369
This commit is contained in:
Fabio Riccardi 2020-05-26 13:58:36 -07:00 committed by TensorFlower Gardener
parent 7280cb3133
commit 7f2d8106f5
2 changed files with 43 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/lite/delegates/gpu/api.cc
View File

@ -31,6 +31,12 @@ struct ObjectTypeGetter {
ObjectType operator()(OpenClTexture) const {
return ObjectType::OPENCL_TEXTURE;
}
ObjectType operator()(VulkanBuffer) const {
return ObjectType::VULKAN_BUFFER;
}
ObjectType operator()(VulkanTexture) const {
return ObjectType::VULKAN_TEXTURE;
}
ObjectType operator()(CpuMemory) const { return ObjectType::CPU_MEMORY; }
};
@ -42,6 +48,8 @@ struct ObjectValidityChecker {
}
bool operator()(OpenClBuffer obj) const { return obj.memobj; }
bool operator()(OpenClTexture obj) const { return obj.memobj; }
bool operator()(VulkanBuffer obj) const { return obj.memory; }
bool operator()(VulkanTexture obj) const { return obj.memory; }
bool operator()(CpuMemory obj) const {
return obj.data != nullptr && obj.size_bytes > 0 &&
(data_type == DataType::UNKNOWN ||
@ -81,6 +89,10 @@ bool IsObjectPresent(ObjectType type, const TensorObject& obj) {
return absl::get_if<OpenClBuffer>(&obj);
case ObjectType::OPENCL_TEXTURE:
return absl::get_if<OpenClTexture>(&obj);
case ObjectType::VULKAN_BUFFER:
return absl::get_if<VulkanBuffer>(&obj);
case ObjectType::VULKAN_TEXTURE:
return absl::get_if<VulkanTexture>(&obj);
case ObjectType::UNKNOWN:
return false;
}

33
tensorflow/lite/delegates/gpu/api.h
View File

@ -71,6 +71,8 @@ enum class ObjectType {
CPU_MEMORY,
OPENCL_TEXTURE,
OPENCL_BUFFER,
VULKAN_BUFFER,
VULKAN_TEXTURE
};
struct OpenGlBuffer {
@ -104,11 +106,37 @@ struct OpenClTexture {
// TODO(akulik): should it specify texture format?
};
struct VulkanBuffer {
VulkanBuffer() = default;
explicit VulkanBuffer(VkBuffer buffer_, VkDeviceSize size_,
VkDeviceMemory memory_, VkDeviceSize offset_)
: buffer(buffer_), size(size_), memory(memory_), offset(offset_) {}
VkBuffer buffer;
VkDeviceSize size;
VkDeviceMemory memory;
VkDeviceSize offset;
};
struct VulkanTexture {
VulkanTexture() = default;
explicit VulkanTexture(VkDeviceMemory new_memory) : memory(new_memory) {}
VkImage image;
VkImageView image_view;
VkFormat format;
VkExtent3D extent;
VkDeviceMemory memory;
VkDeviceSize offset;
};
struct VulkanMemory {
VulkanMemory() = default;
explicit VulkanMemory(VkDeviceMemory new_memory) : memory(new_memory) {}
VkDeviceMemory memory;
VkDeviceSize size;
VkDeviceSize offset;
};
struct CpuMemory {
@ -195,8 +223,9 @@ bool IsValid(const TensorObjectDef& def);
// @return the number of elements in a tensor object.
uint32_t NumElements(const TensorObjectDef& def);
using TensorObject = absl::variant<absl::monostate, OpenGlBuffer, OpenGlTexture,
CpuMemory, OpenClBuffer, OpenClTexture>;
using TensorObject =
absl::variant<absl::monostate, OpenGlBuffer, OpenGlTexture, CpuMemory,
OpenClBuffer, OpenClTexture, VulkanBuffer, VulkanTexture>;
// @return true if object is set and corresponding values are defined.
bool IsValid(const TensorObjectDef& def, const TensorObject& object);