Lines Matching refs:SetTensorInfo

34     input->GetOutputSlot(0).SetTensorInfo(info);
35     add->GetOutputSlot(0).SetTensorInfo(info);
85 …input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float32));
86 …normalize->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float…
122 …input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float32));
123 …normalize->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float…
177     layer->GetOutputSlot(0).SetTensorInfo(desc);
182     normLayer->GetOutputSlot(0).SetTensorInfo(desc);
187     layer->GetOutputSlot(0).SetTensorInfo(desc);
194     layer->GetOutputSlot(0).SetTensorInfo(desc);
201     layer->GetOutputSlot(0).SetTensorInfo(desc);
249     layer->GetOutputSlot(0).SetTensorInfo(desc);
254     normLayer->GetOutputSlot(0).SetTensorInfo(desc);
259     layer->GetOutputSlot(0).SetTensorInfo(desc);
266     layer->GetOutputSlot(0).SetTensorInfo(desc);
273     layer->GetOutputSlot(0).SetTensorInfo(desc);
317 …input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float32));
318 …normalize->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float…
411     weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
414     biasLayer->GetOutputSlot(0).SetTensorInfo(biasesInfo);
420     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
421     convLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);