Home
last modified time | relevance | path

Searched refs:OPENCL_BUFFER (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor_type_util.cc26 return ObjectType::OPENCL_BUFFER; in ToObjectType()
59 case ObjectType::OPENCL_BUFFER: in ToTensorStorageType()
Dgl_interop.h150 output.object_type == ObjectType::OPENCL_BUFFER) || in IsSupported()
151 (input.object_type == ObjectType::OPENCL_BUFFER && in IsSupported()
Dapi.cc104 return (object_type == ObjectType::OPENCL_BUFFER || in IsSupported()
191 case ObjectType::OPENCL_BUFFER: { in MaybeAllocateExternalObject()
270 outer_def.internal_def.object_def.object_type = ObjectType::OPENCL_BUFFER; in MakeOuterInnerDefs()
360 cl_def.external_def.object_def.object_type = ObjectType::OPENCL_BUFFER; in MakeClDef()
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dapi.cc29 return ObjectType::OPENCL_BUFFER; in operator ()()
88 case ObjectType::OPENCL_BUFFER: in IsObjectPresent()
Dapi.h78 OPENCL_BUFFER, enumerator
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc79 def.object_type == ObjectType::OPENCL_BUFFER && in IsBHWCOpenCLBuffer()
84 const bool is_buffer_tensor = def.object_type == ObjectType::OPENCL_BUFFER && in IsOpenCLTensor()
401 return type == ObjectType::OPENCL_BUFFER || in IsOpenClTextureOrBuffer()