FixedReshapeLayer Struct Reference

#include <GraphBuilder.h>

Data Structures
class	Return

Public Member Functions
std::unique_ptr< Return >	operator() (GraphBuilder::Context *ctx)

Detailed Description

Definition at line 481 of file GraphBuilder.h.

Member Function Documentation

operator()()

std::unique_ptr< Return > FixedReshapeLayer::operator() ( GraphBuilder::Context *  ctx )

inline

Definition at line 505 of file GraphBuilder.h.

  {
    auto reshape_node = ctx->graph()->nodes()->create<loco::FixedReshape>();
 
    reshape_node->input(ctx->stack()->pop());
 
    ctx->stack()->push(reshape_node);
 
    return std::make_unique<Return>(reshape_node);
  }

References loco::NodePool::create(), GraphBuilder::Context::graph(), loco::Reshape< ReshapeType::Fixed >::input(), loco::Graph::nodes(), GraphBuilder::Stack::pop(), GraphBuilder::Stack::push(), and GraphBuilder::Context::stack().

---

The documentation for this struct was generated from the following file:

• compiler/loco/src/Service/GraphBuilder.h