• Home
  • Raw
  • Download

Lines Matching refs:weightsTensorInfo

533     armnn::TensorInfo weightsTensorInfo(weightShape, DataType, inputsQScale);  local
534 weightsTensorInfo.SetConstant();
541 weights->m_LayerOutput = std::make_unique<ScopedTensorHandle>(weightsTensorInfo);
546 Connect(weights, layer, weightsTensorInfo, 0, 1);
597 armnn::TensorInfo weightsTensorInfo(weightShape, DataType, inputsQScale); local
598 weightsTensorInfo.SetConstant();
624 weights->m_LayerOutput = std::make_unique<ScopedTensorHandle>(weightsTensorInfo);
631 Connect(weights, layer, weightsTensorInfo, 0, 1);
691 armnn::TensorInfo weightsTensorInfo(weightShape, DataType, inputsQScale); local
692 weightsTensorInfo.SetConstant();
704 Connect(weights, layer, weightsTensorInfo, 0, 1);
1149 armnn::TensorInfo weightsTensorInfo(weightShape, DataType, inputsQScale); in CreateDirectConvolution2dWorkloadTest() local
1150 weightsTensorInfo.SetConstant(); in CreateDirectConvolution2dWorkloadTest()
1160 weights->m_LayerOutput = std::make_unique<ScopedTensorHandle>(weightsTensorInfo); in CreateDirectConvolution2dWorkloadTest()
1167 Connect(weights, layer, weightsTensorInfo, 0, 1); in CreateDirectConvolution2dWorkloadTest()
1263 armnn::TensorInfo weightsTensorInfo({7, 20}, DataType, inputsQScale); in CreateFullyConnectedWorkloadTest() local
1264 weightsTensorInfo.SetConstant(); in CreateFullyConnectedWorkloadTest()
1271 weights->m_LayerOutput = std::make_unique<ScopedTensorHandle>(weightsTensorInfo); in CreateFullyConnectedWorkloadTest()
1276 Connect(weights, layer, weightsTensorInfo, 0, 1); in CreateFullyConnectedWorkloadTest()
1308 armnn::TensorInfo weightsTensorInfo({7, 20}, DataType, inputsQScale); in CreateFullyConnectedWithBlobWorkloadTest() local
1310 weightsTensorInfo.SetConstant(); in CreateFullyConnectedWithBlobWorkloadTest()
1333 weights->m_LayerOutput = std::make_unique<ScopedTensorHandle>(weightsTensorInfo); in CreateFullyConnectedWithBlobWorkloadTest()
1340 Connect(weights, layer, weightsTensorInfo, 0, 1); in CreateFullyConnectedWithBlobWorkloadTest()
2104 TensorInfo weightsTensorInfo(TensorShape({16, 1, 1, 16}), dataType, 0.9f, 0, true); in CreatePreCompiledWorkloadTest() local
2105 unsigned int weightsLength = weightsTensorInfo.GetNumElements(); in CreatePreCompiledWorkloadTest()
2114 armnn::ConstTensor weights(weightsTensorInfo, convWeightsData); in CreatePreCompiledWorkloadTest()