Searched refs:elementwiseBinaryLayer (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | ElementwiseBinary.hpp | 336 armnn::IConnectableLayer* elementwiseBinaryLayer = nullptr; in VisitElementwiseBinaryOperator() local
341 elementwiseBinaryLayer = delegateData.m_Network->AddElementwiseBinaryLayer( in VisitElementwiseBinaryOperator()
345 elementwiseBinaryLayer = delegateData.m_Network->AddElementwiseBinaryLayer( in VisitElementwiseBinaryOperator()
352 elementwiseBinaryLayer = &multiLayer; in VisitElementwiseBinaryOperator()
356 elementwiseBinaryLayer = delegateData.m_Network->AddElementwiseBinaryLayer( in VisitElementwiseBinaryOperator()
360 elementwiseBinaryLayer = delegateData.m_Network->AddElementwiseBinaryLayer( in VisitElementwiseBinaryOperator()
364 elementwiseBinaryLayer = delegateData.m_Network->AddElementwiseBinaryLayer( in VisitElementwiseBinaryOperator()
368 elementwiseBinaryLayer = delegateData.m_Network->AddElementwiseBinaryLayer( in VisitElementwiseBinaryOperator()
374 ARMNN_ASSERT(elementwiseBinaryLayer != nullptr); in VisitElementwiseBinaryOperator()
375 armnn::IOutputSlot& outputSlot = elementwiseBinaryLayer->GetOutputSlot(0); in VisitElementwiseBinaryOperator()
[all …]
|
| /aosp_15_r20/external/armnn/src/backends/backendsCommon/test/ |
| H A D | ElementwiseBinaryEndToEndTestImpl.hpp | 39 …IConnectableLayer* elementwiseBinaryLayer = net->AddElementwiseBinaryLayer(operation, "elementwise… in CreateElementwiseBinaryNetwork() local
42 Connect(input1, elementwiseBinaryLayer, input1TensorInfo, 0, 0); in CreateElementwiseBinaryNetwork()
43 Connect(input2, elementwiseBinaryLayer, input2TensorInfo, 0, 1); in CreateElementwiseBinaryNetwork()
44 Connect(elementwiseBinaryLayer, output, outputTensorInfo, 0, 0); in CreateElementwiseBinaryNetwork()
|
| /aosp_15_r20/external/armnn/src/armnnSerializer/test/ |
| H A D | SerializerTests.cpp | 991 …armnn::IConnectableLayer* const elementwiseBinaryLayer = network->AddElementwiseBinaryLayer(descri… in SerializeElementwiseBinaryTest() local
995 inputLayer0->GetOutputSlot(0).Connect(elementwiseBinaryLayer->GetInputSlot(0)); in SerializeElementwiseBinaryTest()
996 inputLayer1->GetOutputSlot(0).Connect(elementwiseBinaryLayer->GetInputSlot(1)); in SerializeElementwiseBinaryTest()
997 elementwiseBinaryLayer->GetOutputSlot(0).Connect(outputLayer->GetInputSlot(0)); in SerializeElementwiseBinaryTest()
1001 elementwiseBinaryLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo); in SerializeElementwiseBinaryTest()
|