Lines Matching refs:arm_compute
51 arm_compute::ITensor& GetTensor() override { return m_Tensor; } in GetTensor()
52 arm_compute::ITensor const& GetTensor() const override { return m_Tensor; } in GetTensor()
75 virtual arm_compute::DataType GetDataType() const override in GetDataType()
80 …virtual void SetMemoryGroup(const std::shared_ptr<arm_compute::IMemoryGroup>& memoryGroup) override in SetMemoryGroup()
82 m_MemoryGroup = PolymorphicPointerDowncast<arm_compute::MemoryGroup>(memoryGroup); in SetMemoryGroup()
140 arm_compute::Status status = m_Tensor.allocator()->import_memory(memory); in Import()
161 arm_compute::Status status = m_Tensor.allocator()->import_memory(memory); in Import()
190 case arm_compute::DataType::F32: in CopyOutTo()
194 case arm_compute::DataType::U8: in CopyOutTo()
195 case arm_compute::DataType::QASYMM8: in CopyOutTo()
199 case arm_compute::DataType::QSYMM8: in CopyOutTo()
200 case arm_compute::DataType::QASYMM8_SIGNED: in CopyOutTo()
204 case arm_compute::DataType::BFLOAT16: in CopyOutTo()
208 case arm_compute::DataType::F16: in CopyOutTo()
212 case arm_compute::DataType::S16: in CopyOutTo()
213 case arm_compute::DataType::QSYMM16: in CopyOutTo()
217 case arm_compute::DataType::S32: in CopyOutTo()
233 case arm_compute::DataType::F32: in CopyInFrom()
237 case arm_compute::DataType::U8: in CopyInFrom()
238 case arm_compute::DataType::QASYMM8: in CopyInFrom()
242 case arm_compute::DataType::QSYMM8: in CopyInFrom()
243 case arm_compute::DataType::QASYMM8_SIGNED: in CopyInFrom()
244 case arm_compute::DataType::QSYMM8_PER_CHANNEL: in CopyInFrom()
248 case arm_compute::DataType::BFLOAT16: in CopyInFrom()
252 case arm_compute::DataType::F16: in CopyInFrom()
256 case arm_compute::DataType::S16: in CopyInFrom()
257 case arm_compute::DataType::QSYMM16: in CopyInFrom()
261 case arm_compute::DataType::S32: in CopyInFrom()
272 arm_compute::Tensor m_Tensor;
273 std::shared_ptr<arm_compute::MemoryGroup> m_MemoryGroup;
284 const arm_compute::TensorShape& shape, in NeonSubTensorHandle()
285 const arm_compute::Coordinates& coords) in NeonSubTensorHandle()
291 arm_compute::ITensor& GetTensor() override { return m_Tensor; } in GetTensor()
292 arm_compute::ITensor const& GetTensor() const override { return m_Tensor; } in GetTensor()
299 virtual arm_compute::DataType GetDataType() const override in GetDataType()
304 virtual void SetMemoryGroup(const std::shared_ptr<arm_compute::IMemoryGroup>&) override {} in SetMemoryGroup()
328 case arm_compute::DataType::F32: in CopyOutTo()
332 case arm_compute::DataType::U8: in CopyOutTo()
333 case arm_compute::DataType::QASYMM8: in CopyOutTo()
337 case arm_compute::DataType::QSYMM8: in CopyOutTo()
338 case arm_compute::DataType::QASYMM8_SIGNED: in CopyOutTo()
342 case arm_compute::DataType::S16: in CopyOutTo()
343 case arm_compute::DataType::QSYMM16: in CopyOutTo()
347 case arm_compute::DataType::S32: in CopyOutTo()
363 case arm_compute::DataType::F32: in CopyInFrom()
367 case arm_compute::DataType::U8: in CopyInFrom()
368 case arm_compute::DataType::QASYMM8: in CopyInFrom()
372 case arm_compute::DataType::QSYMM8: in CopyInFrom()
373 case arm_compute::DataType::QASYMM8_SIGNED: in CopyInFrom()
377 case arm_compute::DataType::S16: in CopyInFrom()
378 case arm_compute::DataType::QSYMM16: in CopyInFrom()
382 case arm_compute::DataType::S32: in CopyInFrom()
393 arm_compute::SubTensor m_Tensor;