ClTensorDecorator wraps an existing CL tensor allowing us to override the TensorInfo for it.
More...
#include <ClTensorHandle.hpp>
|
| uint8_t * | do_map (cl::CommandQueue &q, bool blocking) override |
| void | do_unmap (cl::CommandQueue &q) override |
ClTensorDecorator wraps an existing CL tensor allowing us to override the TensorInfo for it.
Definition at line 379 of file ClTensorHandle.hpp.
◆ ClTensorDecorator() [1/4]
◆ ClTensorDecorator() [2/4]
Definition at line 25 of file ClTensorHandle.cpp.
26 : m_Original(nullptr), m_TensorInfo()
27 {
28 m_TensorInfo = armcomputetensorutils::BuildArmComputeTensorInfo(tensorInfo);
29 m_Original = original;
30 }
◆ ~ClTensorDecorator()
◆ ClTensorDecorator() [3/4]
◆ ClTensorDecorator() [4/4]
◆ cl_buffer()
| const cl::Buffer & cl_buffer |
( |
| ) |
const |
|
override |
Definition at line 42 of file ClTensorHandle.cpp.
43 {
44 ARM_COMPUTE_ERROR_ON(m_Original == nullptr);
45 return m_Original->cl_buffer();
46 }
◆ do_map()
| uint8_t * do_map |
( |
cl::CommandQueue & | q, |
|
|
bool | blocking ) |
|
overrideprotected |
Definition at line 68 of file ClTensorHandle.cpp.
69 {
70 if(m_Original->buffer() == nullptr)
71 {
72 m_Original->map(q, blocking);
73 }
74 return m_Original->buffer();
75 }
◆ do_unmap()
| void do_unmap |
( |
cl::CommandQueue & | q | ) |
|
|
overrideprotected |
◆ info() [1/2]
| arm_compute::ITensorInfo * info |
( |
| ) |
const |
|
overridevirtual |
◆ info() [2/2]
| arm_compute::ITensorInfo * info |
( |
| ) |
|
|
overridevirtual |
◆ map()
| void map |
( |
bool | blocking = true | ) |
|
Definition at line 58 of file ClTensorHandle.cpp.
59 {
60 arm_compute::ICLTensor::map(arm_compute::CLScheduler::get().queue(), blocking);
61 }
◆ operator=() [1/2]
◆ operator=() [2/2]
◆ parent()
| arm_compute::ICLTensor * parent |
( |
| ) |
|
◆ quantization()
| arm_compute::CLQuantization quantization |
( |
| ) |
const |
|
override |
◆ unmap()
Definition at line 63 of file ClTensorHandle.cpp.
64 {
65 arm_compute::ICLTensor::unmap(arm_compute::CLScheduler::get().queue());
66 }
The documentation for this class was generated from the following files: