Lines Matching refs:SetTensorInfo

43     layer->GetOutputSlot(0).SetTensorInfo(desc);
48     normLayer->GetOutputSlot(0).SetTensorInfo(desc);
53     layer->GetOutputSlot(0).SetTensorInfo(desc);
60     layer->GetOutputSlot(0).SetTensorInfo(desc);
67     layer->GetOutputSlot(0).SetTensorInfo(desc);
112 …input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float32));
113 …permute->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 4, 1, 4 }, armnn::DataType::Float32…
148     input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 4, 3, 2 }, armnn::DataType::Float32));
149     meanLayer->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 2 }, armnn::DataType::Float32));
182     input->GetOutputSlot(0).SetTensorInfo(info);
183     activation->GetOutputSlot(0).SetTensorInfo(info);