11 namespace optimizations
29 const TensorInfo& inInfo = parentOut->GetTensorInfo();
42 const std::string name = std::string(
"merged-") + base.
GetName() + std::string(
"-with-") + child.
GetName();
49 parentOut = &newReshape.GetOutputSlot();
const OutputHandler & GetOutputHandler(unsigned int i=0) const
LayerType GetType() const
const char * GetName() const override
~OptimizeConsecutiveReshapesImpl()=default
This layer represents a reshape operation.
const TensorInfo & GetTensorInfo() const
Gets the matching TensorInfo for the output.
A ReshapeDescriptor for the ReshapeLayer.
unsigned int GetNumConnections() const override
void Run(Graph &graph, InputSlot &connection) const
OptimizeConsecutiveReshapesImpl()=default
void MoveAllConnections(OutputSlot &destination)
Moves all connections to another OutputSlot.
void SetTensorInfo(const TensorInfo &tensorInfo)
Sets the TensorInfo used by this output handler.
LayerT * InsertNewLayer(InputSlot &insertBefore, Args &&... args)
const TensorShape & GetShape() const
Layer & GetOwningLayer() const
const OutputSlot & GetOutputSlot(unsigned int index=0) const override
const InputSlot & GetInputSlot(unsigned int index) const override