#include <GraphTestcase.h>

Collaboration diagram for GraphTestcase< GraphCode::Relu >:

[legend]

Public Member Functions
	GraphTestcase ()

loco::Graph *	graph ()

Data Fields
loco::Pull *	pull_node = nullptr

loco::ReLU *	relu_node = nullptr

loco::Push *	push_node = nullptr

Detailed Description

Definition at line 144 of file GraphTestcase.h.

Constructor & Destructor Documentation

◆ GraphTestcase()

GraphTestcase< GraphCode::Relu >::GraphTestcase ( )

inline

Definition at line 147 of file GraphTestcase.h.

  {
    // Create a sample network
    _graph = loco::make_graph();
 
    auto graph_builder = make_graph_builder(_graph.get());
 
    pull_node = graph_builder->push<InputLayer>()->name("input")->node();
    relu_node = graph_builder->push<ReLULayer>()->node();
    push_node = graph_builder->push<OutputLayer>()->name("output")->node();
  }