Home
last modified time | relevance | path

Searched refs:m_KernelTensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonConvolution2dWorkload.cpp92 m_KernelTensor = std::make_unique<arm_compute::Tensor>(); in NeonConvolution2dWorkload()
93 …BuildArmComputeTensor(*m_KernelTensor, info.m_InputTensorInfos[1], m_Data.m_Parameters.m_DataLayou… in NeonConvolution2dWorkload()
109 m_KernelTensor.get(), in NeonConvolution2dWorkload()
120 m_KernelTensor->info(), in NeonConvolution2dWorkload()
164 InitializeArmComputeTensorData(*m_KernelTensor, m_KernelTensorInfo, m_Data.m_Inputs[1]); in Execute()
171 FreeTensorIfUnused(m_KernelTensor); in Execute()
H A DNeonTransposeConvolution2dWorkload.cpp69 m_KernelTensor = std::make_unique<arm_compute::Tensor>(); in NeonTransposeConvolution2dWorkload()
70 …BuildArmComputeTensor(*m_KernelTensor, m_Data.m_Weight->GetTensorInfo(), m_Data.m_Parameters.m_Dat… in NeonTransposeConvolution2dWorkload()
98 m_Layer->configure(&input, m_KernelTensor.get(), m_BiasTensor.get(), &output, padStrideInfo); in NeonTransposeConvolution2dWorkload()
102 InitializeArmComputeTensorData(*m_KernelTensor, m_Data.m_Weight); in NeonTransposeConvolution2dWorkload()
121 FreeTensorIfUnused(m_KernelTensor); in FreeUnusedTensors()
H A DNeonTransposeConvolution2dWorkload.hpp39 std::unique_ptr<arm_compute::Tensor> m_KernelTensor; member in armnn::NeonTransposeConvolution2dWorkload
H A DNeonConvolution2dWorkload.hpp44 mutable std::unique_ptr<arm_compute::Tensor> m_KernelTensor; member in armnn::NeonConvolution2dWorkload