ONE - On-device Neural Engine
Loading...
Searching...
No Matches
DepthwiseConv2DLayer Struct Referencefinal

#include <GraphBuilder.h>

Data Structures

class  Return
 

Public Member Functions

std::unique_ptr< Returnoperator() (GraphBuilder::Context *ctx)
 

Detailed Description

Definition at line 421 of file GraphBuilder.h.

Member Function Documentation

◆ operator()()

std::unique_ptr< Return > DepthwiseConv2DLayer::operator() ( GraphBuilder::Context ctx)
inline

Definition at line 438 of file GraphBuilder.h.

439 {
440 auto depthwiseconv2d_node = ctx->graph()->nodes()->create<loco::DepthwiseConv2D>();
441
442 depthwiseconv2d_node->ker(ctx->stack()->pop());
443 depthwiseconv2d_node->ifm(ctx->stack()->pop());
444
445 ctx->stack()->push(depthwiseconv2d_node);
446
447 return std::make_unique<Return>(depthwiseconv2d_node);
448 }
loco::Graph * graph(void)
Stack * stack(void)
void push(loco::Node *node)
loco::Node * pop(void)
Depthwise 2D Convolution.
Definition Nodes.h:582
Node * ker(void) const
Definition Nodes.h:587
NodeContext * nodes(void)
Definition Graph.h:218
Derived * create(Args &&...args)
Definition NodePool.h:37

References loco::NodePool::create(), GraphBuilder::Context::graph(), loco::DepthwiseConv2D::ker(), loco::Graph::nodes(), GraphBuilder::Stack::pop(), GraphBuilder::Stack::push(), and GraphBuilder::Context::stack().


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