#include <GraphTestcase.h>

Collaboration diagram for GraphTestcase< GraphCode::FixedReshape >:

[legend]

Public Member Functions
	GraphTestcase ()

loco::Graph *	graph ()

Data Fields
loco::Pull *	pull_node = nullptr

loco::FixedReshape *	reshape_node = nullptr

loco::Push *	push_node = nullptr

Detailed Description

Definition at line 445 of file GraphTestcase.h.

Constructor & Destructor Documentation

◆ GraphTestcase()

GraphTestcase< GraphCode::FixedReshape >::GraphTestcase ( )

inline

Definition at line 448 of file GraphTestcase.h.

  {
    _graph = loco::make_graph();
 
    auto graph_builder = make_graph_builder(_graph.get());
 
    pull_node = graph_builder->push<InputLayer>()->name("input")->node();
    reshape_node = graph_builder->push<FixedReshapeLayer>()->node();
    push_node = graph_builder->push<OutputLayer>()->name("output")->node();
  }