ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Member Functions
FixedReshapeLayer::Return Class Reference

#include <GraphBuilder.h>

Public Member Functions

 Return (loco::FixedReshape *node)
 
Return * shape (std::initializer_list< uint32_t > dims)
 
loco::FixedReshape * node (void)
 

Detailed Description

Definition at line 483 of file GraphBuilder.h.

Constructor & Destructor Documentation

◆ Return()

FixedReshapeLayer::Return::Return ( loco::FixedReshape *  node)
inline

Definition at line 486 of file GraphBuilder.h.

486 : _node{node}
487 {
488 // DO NOTHING
489 }
FixedReshapeLayer::Return::node
loco::FixedReshape * node(void)
Definition GraphBuilder.h:499

Member Function Documentation

◆ node()

loco::FixedReshape * FixedReshapeLayer::Return::node ( void  )
inline

Definition at line 499 of file GraphBuilder.h.

499{ return _node; }

◆ shape()

Return * FixedReshapeLayer::Return::shape ( std::initializer_list< uint32_t >  dims)
inline

Definition at line 492 of file GraphBuilder.h.

493 {
494 _node->shape(dims);
495 return this;
496 }

Referenced by RandomDataGenerator.RandomDataGenerator::_gen_float32(), RandomDataGenerator.RandomDataGenerator::_gen_int16(), and RandomDataGenerator.RandomDataGenerator::_gen_uint8().

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