Home
last modified time | relevance | path

Searched refs:GetOutputHandler (Results 1 – 21 of 21) sorted by relevance

/external/armnn/src/armnn/test/optimizations/
DMoveTransposeUpTests.cpp34 head->GetOutputHandler().SetTensorInfo(transposed);
38 head->GetOutputHandler().SetTensorInfo(info);
41 head->GetOutputHandler().SetTensorInfo(info);
45 ->GetOutputHandler()
50 head->GetOutputHandler().SetTensorInfo(info);
53 head->GetOutputHandler().SetTensorInfo(info);
56 head->GetOutputHandler().SetTensorInfo(info);
59 head->GetOutputHandler().SetTensorInfo(info);
63 ->GetOutputHandler()
68 ->GetOutputHandler()
DMovePermuteUpTests.cpp33 head->GetOutputHandler().SetTensorInfo(permuted);
37 head->GetOutputHandler().SetTensorInfo(info);
40 head->GetOutputHandler().SetTensorInfo(info);
44 ->GetOutputHandler()
49 head->GetOutputHandler().SetTensorInfo(info);
52 head->GetOutputHandler().SetTensorInfo(info);
55 head->GetOutputHandler().SetTensorInfo(info);
58 head->GetOutputHandler().SetTensorInfo(info);
62 ->GetOutputHandler()
67 ->GetOutputHandler()
DOptimizeConsecutiveReshapesTests.cpp25 input->GetOutputHandler().SetTensorInfo(info0);
40 reshape1->GetOutputHandler().SetTensorInfo(info1);
41 reshape2->GetOutputHandler().SetTensorInfo(info2);
53 … (reshapeLayer->GetOutputHandler().GetTensorInfo().GetShape() == info2.GetShape()); in __anonc43624d30102()
71 reshapeToIn->GetOutputHandler().SetTensorInfo(info0);
DPermuteAsReshapeTests.cpp30 ->GetOutputHandler()
37 ->GetOutputHandler()
51 (reshapeLayer->GetOutputHandler().GetTensorInfo().GetShape() == infoOut.GetShape()); in __anone788bd1d0102()
DTransposeAsReshapeTests.cpp30 ->GetOutputHandler()
37 ->GetOutputHandler()
51 (reshapeLayer->GetOutputHandler().GetTensorInfo().GetShape() == infoOut.GetShape()); in __anonf2b7e81a0102()
DPermuteAndBatchToSpaceAsDepthToSpaceTests.cpp104 … layer->GetOutputHandler().GetTensorInfo() == TensorInfo({ 1, 4, 6, 1 }, DataType::Float32); in __anonf0bff9fd0202()
134 … layer->GetOutputHandler().GetTensorInfo() == TensorInfo({ 1, 4, 6, 1 }, DataType::Float32); in __anonf0bff9fd0302()
/external/armnn/src/armnn/
DLayer.cpp67 layer.GetOutputHandler().SetTensorInfo(tensorInfo); in Insert()
89 GetOutputHandler().SetTensorInfo(tensorInfo); in SetTensorInfo()
94 return GetOutputHandler().GetTensorInfo(); in GetTensorInfo()
103 return GetOutputHandler().IsTensorInfoSet(); in IsTensorInfoSet()
155 destination.GetOutputHandler().SetTensorInfo(GetOutputHandler().GetTensorInfo()); in MoveAllConnections()
261 const OutputHandler& outputHandler = inputSlot.GetConnectedOutputSlot()->GetOutputHandler(); in CollectWorkloadInputs()
289 OutputHandler& handler = GetOutputHandler(idx); in CreateTensorHandles()
DLayer.hpp126 const OutputHandler& GetOutputHandler() const { return m_OutputHandler; } in GetOutputHandler() function in armnn::OutputSlot
127 OutputHandler& GetOutputHandler() { return m_OutputHandler; } in GetOutputHandler() function in armnn::OutputSlot
232 const OutputHandler& GetOutputHandler(unsigned int i = 0) const in GetOutputHandler() function in armnn::Layer
237 OutputHandler& GetOutputHandler(unsigned int i = 0) in GetOutputHandler() function in armnn::Layer
239 return const_cast<OutputHandler&>(const_cast<const Layer*>(this)->GetOutputHandler(i)); in GetOutputHandler()
DLoadedNetwork.cpp411 layer->GetOutputSlot(0).GetOutputHandler().GetData(); in LoadedNetwork()
638 ITensorHandle* tensorHandle = outSlot.GetOutputHandler().GetData(); in AllocateAndExecuteConstantWorkloadsAsync()
895 … OutputHandler& handler = const_cast<OutputHandler&>(inputLayer->GetOutputHandler(0)); in EnqueueWorkload()
973 … outputLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetOutputHandler(); in EnqueueWorkload()
1050 const OutputHandler& handler = layer.GetOutputHandler(); in EnqueueInput()
1117 …tputHandler& outputHandler = layer.GetInputSlots()[0].GetConnectedOutputSlot()->GetOutputHandler(); in EnqueueOutput()
2267 size_t alignedSize = align(outputSlot.GetOutputHandler().GetTensorInfo().GetNumBytes()); in CreateMemoryProfileAsync()
2390 ITensorHandle* tensorHandle = outputSlot.GetOutputHandler().GetData(); in CreateMemoryProfile()
2399 … size_t alignedSize = align(outputSlot.GetOutputHandler().GetTensorInfo().GetNumBytes()); in CreateMemoryProfile()
2446 ITensorHandle* tensorHandle = outputSlot->GetOutputHandler().GetData(); in CreateMemoryProfile()
DGraph.cpp216 … ITensorHandle *tensorHandle = TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()); in AllocateDynamicBuffers()
234 … ITensorHandle *tensorHandle = TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()); in AllocateDynamicBuffers()
261 slot->GetConnectedOutputSlot()->GetOutputHandler().GetData()); in AllocateDynamicBuffers()
/external/armnn/src/armnn/optimizations/
DPermuteAsReshape.hpp22 const TensorInfo& outInfo = permute.GetOutputHandler().GetTensorInfo(); in Run()
41 const TensorShape& outShape = layer.GetOutputHandler().GetTensorInfo().GetShape(); in IsReshape()
DTransposeAsReshape.hpp22 const TensorInfo& outInfo = transpose.GetOutputHandler().GetTensorInfo(); in Run()
41 const TensorShape& outShape = layer.GetOutputHandler().GetTensorInfo().GetShape(); in IsReshape()
DOptimizeConsecutiveReshapes.hpp30 const TensorInfo& outInfo = child.GetOutputHandler().GetTensorInfo(); in Run()
DMoveTransposeUp.hpp46 permLayer.GetOutputHandler().SetTensorInfo(permOutInfo); in Run()
DMovePermuteUp.hpp46 permLayer.GetOutputHandler().SetTensorInfo(permOutInfo); in Run()
/external/armnn/src/backends/backendsCommon/test/
DIsLayerSupportedTestImpl.hpp987 input->GetOutputHandler(0).SetTensorInfo(inputTensorInfo); in IsConvertLayerSupportedTests()
989 layer->GetOutputHandler(0).SetTensorInfo(outputTensorInfo); in IsConvertLayerSupportedTests()
1017 input0->GetOutputHandler(0).SetTensorInfo(inputTensorInfo0); in IsLogicalBinaryLayerSupportedTests()
1018 input1->GetOutputHandler(0).SetTensorInfo(inputTensorInfo1); in IsLogicalBinaryLayerSupportedTests()
1021 layer->GetOutputHandler(0).SetTensorInfo(outputTensorInfo); in IsLogicalBinaryLayerSupportedTests()
1049 input0->GetOutputHandler(0).SetTensorInfo(inputTensorInfo0); in IsLogicalBinaryLayerBroadcastSupportedTests()
1050 input1->GetOutputHandler(0).SetTensorInfo(inputTensorInfo1); in IsLogicalBinaryLayerBroadcastSupportedTests()
1053 layer->GetOutputHandler(0).SetTensorInfo(outputTensorInfo); in IsLogicalBinaryLayerBroadcastSupportedTests()
1076 input->GetOutputHandler(0).SetTensorInfo(inputTensorInfo); in IsMeanLayerSupportedTests()
1078 layer->GetOutputHandler(0).SetTensorInfo(outputTensorInfo); in IsMeanLayerSupportedTests()
[all …]
/external/armnn/src/armnn/layers/
DConcatLayer.cpp65 ITensorHandle* parentTensor = currentLayer->GetOutputHandler(0).GetData(); in CreateTensors()
66 const TensorInfo& parentInfo = currentLayer->GetOutputHandler(0).GetTensorInfo(); in CreateTensors()
163 OutputHandler& outputHandler = slot->GetOutputHandler(); in CreateTensors()
DInputLayer.cpp34 ConditionalThrow<LayerValidationException>(GetOutputHandler(0).IsTensorInfoSet(), in ValidateTensorShapesFromInputs()
DSplitterLayer.cpp49 …nst OutputHandler& outputHandler = GetInputSlots()[0].GetConnectedOutputSlot()->GetOutputHandler(); in CreateTensors()
/external/armnn/src/backends/neon/test/
DNeonTensorHandleTests.cpp156 if (TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()))
239 if (TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()))
408 if (TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()))
585 if (TraceSubTensorHandleAncestry(slot->GetOutputHandler().GetData()))
/external/armnn/src/armnn/test/
DOptimizerTests.cpp343 head->GetOutputHandler().SetTensorInfo(info);
346 ->GetOutputHandler().SetTensorInfo(info);
349 head->GetOutputHandler().SetTensorInfo(info);
352 head->GetOutputHandler().SetTensorInfo(info);
355 ->GetOutputHandler().SetTensorInfo(info);