| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_context.h | 29 class CLContext {
31 CLContext() {} in CLContext() function
32 CLContext(cl_context context, bool has_ownership);
35 CLContext(CLContext&& context);
36 CLContext& operator=(CLContext&& context);
37 CLContext(const CLContext&) = delete;
38 CLContext& operator=(const CLContext&) = delete;
40 ~CLContext();
54 absl::Status CreateCLContext(const CLDevice& device, CLContext* result);
58 CLContext* result);
|
| D | cl_context.cc | 86 CLContext* result) { in CreateCLContext()
98 *result = CLContext(context, true); in CreateCLContext()
104 CLContext::CLContext(cl_context context, bool has_ownership) in CLContext() function in tflite::gpu::cl::CLContext
107 CLContext::CLContext(CLContext&& context) in CLContext() function in tflite::gpu::cl::CLContext
112 CLContext& CLContext::operator=(CLContext&& context) { in operator =()
121 CLContext::~CLContext() { Release(); } in ~CLContext()
123 void CLContext::Release() { in Release()
130 bool CLContext::IsFloatTexture2DSupported(int num_channels, DataType data_type, in IsFloatTexture2DSupported()
143 absl::Status CreateCLContext(const CLDevice& device, CLContext* result) { in CreateCLContext()
150 CLContext* result) { in CreateCLGLContext()
|
| D | tensor.h | 96 CLContext* context);
120 absl::Status AllocateTensorMemory(const CLContext& context, const BHWC& shape,
124 absl::Status AllocateTensorMemory(const CLContext& context, const BHWDC& shape,
128 absl::Status CreateTensor(const CLContext& context, const BHWC& shape,
131 absl::Status CreateTensor(const CLContext& context, const BHWDC& shape,
134 absl::Status CreateSharedTensor(const CLContext& context, cl_mem memory,
139 absl::Status CreateSharedTensor(const CLContext& context, cl_mem memory,
|
| D | texture2d.cc | 25 CLContext* context, Texture2D* result) { in CreateTexture2D()
83 const Texture2DDescriptor& desc, CLContext* context) { in CreateFromTexture2DDescriptor()
95 absl::Status CreateTexture2DRGBA32F(int width, int height, CLContext* context, in CreateTexture2DRGBA32F()
102 absl::Status CreateTexture2DRGBA16F(int width, int height, CLContext* context, in CreateTexture2DRGBA16F()
109 CLContext* context, Texture2D* result) { in CreateTexture2DRGBA()
114 void* data, CLContext* context, in CreateTexture2DRGBA()
|
| D | buffer.cc | 29 const void* data, CLContext* context, in CreateBuffer()
78 CLContext* context) { in CreateFromBufferDescriptor()
88 absl::Status CreateReadOnlyBuffer(size_t size_in_bytes, CLContext* context, in CreateReadOnlyBuffer()
94 CLContext* context, Buffer* result) { in CreateReadOnlyBuffer()
98 absl::Status CreateReadWriteBuffer(size_t size_in_bytes, CLContext* context, in CreateReadWriteBuffer()
|
| D | texture2d.h | 64 CLContext* context);
78 absl::Status CreateTexture2DRGBA32F(int width, int height, CLContext* context,
82 absl::Status CreateTexture2DRGBA16F(int width, int height, CLContext* context,
86 CLContext* context, Texture2D* result);
89 void* data, CLContext* context,
|
| D | buffer.h | 66 CLContext* context);
75 absl::Status CreateReadOnlyBuffer(size_t size_in_bytes, CLContext* context,
79 CLContext* context, Buffer* result);
81 absl::Status CreateReadWriteBuffer(size_t size_in_bytes, CLContext* context,
|
| D | environment.h | 37 explicit Environment(CLDevice&& device, CLContext&& context,
49 CLContext& context() { return context_; } in context()
72 CLContext context_;
|
| D | program_cache.h | 47 const CLContext& context, const CLDevice& device, CLKernel* result);
51 const CLContext& context,
54 absl::Status AddSerializedCache(const CLContext& context,
|
| D | inference_context.h | 120 absl::Status AllocateMemory(CLContext* context);
122 absl::Status AllocateMemoryForConstTensors(CLContext* context);
124 absl::Status AllocateMemoryForVariableTensors(CLContext* context);
126 absl::Status AllocateMemoryForBuffers(CLContext* context);
128 absl::Status AllocateMemoryForStrongShapes(CLContext* context);
|
| D | cl_program.h | 71 const CLContext& context, const CLDevice& device,
74 absl::Status CreateCLProgramFromBinary(const CLContext& context,
|
| D | tensor.cc | 33 absl::Status AllocateTensorMemory(const CLContext& context, const BHWDC& shape, in AllocateTensorMemory()
192 absl::Status CreateImageBufferFromBuffer(const CLContext& context, in CreateImageBufferFromBuffer()
216 absl::Status CreateTensor(const CLContext& context, const BHWDC& shape, in CreateTensor()
239 absl::Status CreateTensorShared(const CLContext& context, const BHWDC& shape, in CreateTensorShared()
602 CLContext* context) { in CreateFromDescriptor()
624 absl::Status CreateTensor(const CLContext& context, const BHWC& shape, in CreateTensor()
630 absl::Status CreateTensor(const CLContext& context, const BHWDC& shape, in CreateTensor()
635 absl::Status CreateSharedTensor(const CLContext& context, cl_mem memory, in CreateSharedTensor()
643 absl::Status CreateSharedTensor(const CLContext& context, cl_mem memory, in CreateSharedTensor()
650 absl::Status AllocateTensorMemory(const CLContext& context, const BHWC& shape, in AllocateTensorMemory()
[all …]
|
| D | program_cache.cc | 62 const CLContext& context, const CLDevice& device, CLKernel* result) { in GetOrCreateCLKernel()
80 const CLContext& context, in GetOrCreateCLKernel()
87 const CLContext& context, const CLDevice& device, in AddSerializedCache()
|
| D | cl_command_queue.h | 121 const CLContext& context,
125 const CLContext& context,
|
| D | cl_arguments.h | 39 CLContext* context, Arguments* args, std::string* code);
41 CLContext* context);
66 absl::Status AllocateObjects(const Arguments& args, CLContext* context);
|
| D | gl_interop.h | 63 CLContext* context, CLMemory* memory);
69 CLContext* context, CLMemory* memory);
|
| D | environment.cc | 34 CLContext context; in CreateEnvironment()
86 Environment::Environment(CLDevice&& device, CLContext&& context, in Environment()
245 CLContext context; in CreateEnvironment()
|
| D | serialization.h | 34 absl::Status Decode(CLContext* context,
|
| D | linear_storage.h | 55 const TensorLinearDescriptor& desc, CLContext* context);
|
| D | cl_program.cc | 167 const CLContext& context, const CLDevice& device, in CreateCLProgram()
185 absl::Status CreateCLProgramFromBinary(const CLContext& context, in CreateCLProgramFromBinary()
|
| D | cl_operation.h | 37 CLContext* context;
|
| D | gl_interop.cc | 97 CLContext* context, CLMemory* memory) { in CreateClMemoryFromGlBuffer()
113 CLContext* context, CLMemory* memory) { in CreateClMemoryFromGlTexture()
|
| D | linear_storage.cc | 74 const TensorLinearDescriptor& desc, CLContext* context) { in CreateFromTensorLinearDescriptor()
|
| D | inference_context.cc | 485 absl::Status InferenceContext::AllocateMemory(CLContext* context) { in AllocateMemory()
494 CLContext* context) { in AllocateMemoryForConstTensors()
503 CLContext* context) { in AllocateMemoryForVariableTensors()
521 absl::Status InferenceContext::AllocateMemoryForBuffers(CLContext* context) { in AllocateMemoryForBuffers()
575 CLContext* context) { in AllocateMemoryForStrongShapes()
|
| D | cl_command_queue.cc | 283 const CLContext& context, in CreateCLCommandQueue()
314 const CLContext& context, in CreateProfilingCommandQueue()
|