20.11
|
#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 26 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 30 of file AddBroadcastReshapeLayer.hpp.
References AddBroadcastReshapeLayerImpl::AddBroadcastReshapeLayerImpl(), InputSlot::GetConnectedOutputSlot(), Layer::GetInputSlot(), Layer::GetNameStr(), TensorInfo::GetNumDimensions(), Layer::GetOutputSlot(), TensorInfo::GetShape(), OutputSlot::GetTensorInfo(), Layer::GetType(), Graph::InsertNewLayer(), OutputSlot::IsTensorInfoSet(), TensorInfo::SetShape(), OutputSlot::SetTensorInfo(), and AddBroadcastReshapeLayerImpl::~AddBroadcastReshapeLayerImpl().