Lines Matching refs:arm_compute
47 arm_compute::CLTensor& GetTensor() override { return m_Tensor; } in GetTensor()
48 arm_compute::CLTensor const& GetTensor() const override { return m_Tensor; } in GetTensor()
79 const_cast<arm_compute::CLTensor*>(&m_Tensor)->map(blocking); in Map()
83 virtual void Unmap() const override { const_cast<arm_compute::CLTensor*>(&m_Tensor)->unmap(); } in Unmap()
87 virtual arm_compute::DataType GetDataType() const override in GetDataType()
92 …virtual void SetMemoryGroup(const std::shared_ptr<arm_compute::IMemoryGroup>& memoryGroup) override in SetMemoryGroup()
94 m_MemoryGroup = PolymorphicPointerDowncast<arm_compute::MemoryGroup>(memoryGroup); in SetMemoryGroup()
147 case arm_compute::DataType::F32: in CopyOutTo()
151 case arm_compute::DataType::U8: in CopyOutTo()
152 case arm_compute::DataType::QASYMM8: in CopyOutTo()
156 case arm_compute::DataType::QSYMM8: in CopyOutTo()
157 case arm_compute::DataType::QSYMM8_PER_CHANNEL: in CopyOutTo()
158 case arm_compute::DataType::QASYMM8_SIGNED: in CopyOutTo()
162 case arm_compute::DataType::F16: in CopyOutTo()
166 case arm_compute::DataType::S16: in CopyOutTo()
167 case arm_compute::DataType::QSYMM16: in CopyOutTo()
171 case arm_compute::DataType::S32: in CopyOutTo()
189 case arm_compute::DataType::F32: in CopyInFrom()
193 case arm_compute::DataType::U8: in CopyInFrom()
194 case arm_compute::DataType::QASYMM8: in CopyInFrom()
198 case arm_compute::DataType::F16: in CopyInFrom()
202 case arm_compute::DataType::S16: in CopyInFrom()
203 case arm_compute::DataType::QSYMM8: in CopyInFrom()
204 case arm_compute::DataType::QSYMM8_PER_CHANNEL: in CopyInFrom()
205 case arm_compute::DataType::QASYMM8_SIGNED: in CopyInFrom()
209 case arm_compute::DataType::QSYMM16: in CopyInFrom()
213 case arm_compute::DataType::S32: in CopyInFrom()
225 arm_compute::CLTensor m_Tensor;
226 std::shared_ptr<arm_compute::MemoryGroup> m_MemoryGroup;
236 const arm_compute::TensorShape& shape, in ClSubTensorHandle()
237 const arm_compute::Coordinates& coords) in ClSubTensorHandle()
243 arm_compute::CLSubTensor& GetTensor() override { return m_Tensor; } in GetTensor()
244 arm_compute::CLSubTensor const& GetTensor() const override { return m_Tensor; } in GetTensor()
251 const_cast<arm_compute::CLSubTensor*>(&m_Tensor)->map(blocking); in Map()
254 … virtual void Unmap() const override { const_cast<arm_compute::CLSubTensor*>(&m_Tensor)->unmap(); } in Unmap()
258 virtual arm_compute::DataType GetDataType() const override in GetDataType()
263 virtual void SetMemoryGroup(const std::shared_ptr<arm_compute::IMemoryGroup>&) override {} in SetMemoryGroup()
282 case arm_compute::DataType::F32: in CopyOutTo()
286 case arm_compute::DataType::U8: in CopyOutTo()
287 case arm_compute::DataType::QASYMM8: in CopyOutTo()
291 case arm_compute::DataType::F16: in CopyOutTo()
295 case arm_compute::DataType::QSYMM8: in CopyOutTo()
296 case arm_compute::DataType::QSYMM8_PER_CHANNEL: in CopyOutTo()
297 case arm_compute::DataType::QASYMM8_SIGNED: in CopyOutTo()
301 case arm_compute::DataType::S16: in CopyOutTo()
302 case arm_compute::DataType::QSYMM16: in CopyOutTo()
306 case arm_compute::DataType::S32: in CopyOutTo()
324 case arm_compute::DataType::F32: in CopyInFrom()
328 case arm_compute::DataType::U8: in CopyInFrom()
329 case arm_compute::DataType::QASYMM8: in CopyInFrom()
333 case arm_compute::DataType::F16: in CopyInFrom()
337 case arm_compute::DataType::QSYMM8: in CopyInFrom()
338 case arm_compute::DataType::QSYMM8_PER_CHANNEL: in CopyInFrom()
339 case arm_compute::DataType::QASYMM8_SIGNED: in CopyInFrom()
343 case arm_compute::DataType::S16: in CopyInFrom()
344 case arm_compute::DataType::QSYMM16: in CopyInFrom()
348 case arm_compute::DataType::S32: in CopyInFrom()
360 mutable arm_compute::CLSubTensor m_Tensor;