Searched refs:weightsTensor (Results 1 – 9 of 9) sorted by relevance
54 ConstTensor weightsTensor; in Run() local61 weightsTensor = ConstTensor(weightLayer->m_LayerOutput->GetTensorInfo(), in Run()65 auto weightsShape = weightsTensor.GetInfo().GetShape(); in Run()74 const auto* weightsBuffer = static_cast<const T*>(weightsTensor.GetMemoryArea()); in Run()80 … std::vector<T> weightsVector (weightsBuffer, weightsBuffer + weightsTensor.GetNumElements()); in Run()126 ConstTensor fusedWeightsTensor(weightsTensor.GetInfo(), fusedWeightsVector); in Run()
91 armnn::ConstTensor weightsTensor(weightsInfo, &expectedWeightsData); variable93 weights->m_LayerOutput = std::make_unique<armnn::ScopedTensorHandle>(weightsTensor);
53 … weightsTensor:weightTransposeTensor113 weightsTensor:weightTransposeTensor
176 auto weightsTensor = CreateConstTensor(&tfLiteWeightsTensor, in VisitFullyConnectedOperator() local179 … armnn::IConnectableLayer* weightsLayer = delegateData.m_Network->AddConstantLayer(weightsTensor); in VisitFullyConnectedOperator()
58 weightsTensor:(MPSGraphTensor * _Nonnull) weights63 … weightsTensor:(MPSGraphTensor * _Nonnull) weights
264 weightsTensor:weightTensor285 weightsTensor:weightTransposeTensor301 weightsTensor:weightTensor472 weightsTensor:weightTensor494 … weightsTensor:weightTransposeTensor511 weightsTensor:weightTensor
71 ScopedTensorHandle weightsTensor(weights.first); in TransposeConvolution2dTestImpl() local75 queueDescriptor.m_Weight = &weightsTensor; in TransposeConvolution2dTestImpl()77 AllocateAndCopyDataToITensorHandle(&weightsTensor, weights.second.data()); in TransposeConvolution2dTestImpl()
77 …[mpsCD instancesRespondToSelector:@selector(convolution3DWithSourceTensor:weightsTensor:descriptor…
1525 armnn::ConstTensor weightsTensor = ToConstTensor(flatBufferLayer->weights()); in ParseConvolution2d() local1526 auto weightsLayer = m_Network->AddConstantLayer(weightsTensor); in ParseConvolution2d()1528 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsTensor.GetInfo()); in ParseConvolution2d()2231 armnn::ConstTensor weightsTensor = ToConstTensor(flatBufferLayer->weights()); in ParseFullyConnected() local2232 auto weightsLayer = m_Network->AddConstantLayer(weightsTensor); in ParseFullyConnected()2234 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsTensor.GetInfo()); in ParseFullyConnected()