21.05
|
#include <AddBroadcastReshapeLayer.hpp>
Public Member Functions | |
void | Run (Graph &graph, Layer &layer) const |
Run for every ElementwiseBaseLayer. Add Broadcast reshape layer if the inputs shape are different. More... | |
Protected Member Functions | |
AddBroadcastReshapeLayerImpl ()=default | |
~AddBroadcastReshapeLayerImpl ()=default | |
Definition at line 22 of file AddBroadcastReshapeLayer.hpp.
|
protecteddefault |
Referenced by AddBroadcastReshapeLayerImpl::Run().
|
protecteddefault |
Referenced by AddBroadcastReshapeLayerImpl::Run().
Run for every ElementwiseBaseLayer. Add Broadcast reshape layer if the inputs shape are different.
Definition at line 26 of file AddBroadcastReshapeLayer.hpp.
References AddBroadcastReshapeLayerImpl::AddBroadcastReshapeLayerImpl(), armnn::Constant, InputSlot::GetConnectedOutputSlot(), Layer::GetInputSlot(), Layer::GetNameStr(), OutputSlot::GetNumConnections(), TensorInfo::GetNumDimensions(), Layer::GetOutputSlot(), OutputSlot::GetOwningLayer(), TensorInfo::GetShape(), OutputSlot::GetTensorInfo(), Layer::GetType(), Graph::InsertNewLayer(), OutputSlot::IsTensorInfoSet(), ConstantLayer::m_LayerOutput, TensorInfo::SetShape(), OutputSlot::SetTensorInfo(), and AddBroadcastReshapeLayerImpl::~AddBroadcastReshapeLayerImpl().