Lines Matching refs:tensorInfo
21 …tHandle::LayerInputHandle(bool valid, armnn::IOutputSlot* outputSlot, armnn::TensorInfo tensorInfo) in LayerInputHandle() argument
24 , m_TensorInfo(tensorInfo) in LayerInputHandle()
82 ConstTensorPin::ConstTensorPin(armnn::TensorInfo& tensorInfo, in ConstTensorPin() argument
89 if (tensorInfo.GetNumBytes() != numBytes) in ConstTensorPin()
97 m_SwizzledTensorData.resize(tensorInfo.GetNumBytes()); in ConstTensorPin()
98 … SwizzleAndroidNn4dTensorToArmNn(tensorInfo, valueStart, m_SwizzledTensorData.data(), mappings); in ConstTensorPin()
100 m_ConstTensor = armnn::ConstTensor(tensorInfo, m_SwizzledTensorData.data()); in ConstTensorPin()
104 m_ConstTensor = armnn::ConstTensor(tensorInfo, valueStart); in ConstTensorPin()
137 armnn::IConnectableLayer* ProcessActivation(const armnn::TensorInfo& tensorInfo, in ProcessActivation() argument
147 prevLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo); in ProcessActivation()
201 tensorInfo, in ProcessActivation()
212 activationLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo); in ProcessActivation()