Home
last modified time | relevance | path

Searched refs:image2d_arrays (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dgpu_object_desc.h76 std::vector<std::pair<std::string, GPUImage2DArrayDescriptor>> image2d_arrays; member
91 for (const auto& obj : image2d_arrays) { in GetNames()
113 for (const auto& t : image2d_arrays) { in GetReadImagesCount()
138 for (const auto& t : image2d_arrays) { in GetWriteImagesCount()
H A Dtensor_desc.cc334 resources.image2d_arrays.push_back({"image2d_array", desc}); in GetGPUResources()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dgpu_object.h40 std::vector<std::pair<std::string, cl_mem>> image2d_arrays; member
H A Dcl_arguments.cc304 for (const auto& r : resources.image2d_arrays) { in AddGPUResources()
449 for (const auto& r : resources.image2d_arrays) { in SetGPUResources()
H A Dtensor.cc354 resources->image2d_arrays.push_back({"image2d_array", memory_}); in GetGPUResources()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dgpu_object.h44 std::vector<std::pair<std::string, id<MTLTexture>>> image2d_arrays; member
H A Dmetal_arguments.cc626 for (const auto& r : resources.image2d_arrays) { in SetGPUResources()
672 for (const auto& r : resources.image2d_arrays) { in AddGPUResources()
H A Dmetal_spatial_tensor.cc235 resources->image2d_arrays.push_back({"image2d_array", texture_mem_}); in GetGPUResources()