GraphTestcase< GraphCode::FeatureCodec > Class Reference

#include <GraphTestcase.h>

Collaboration diagram for GraphTestcase< GraphCode::FeatureCodec >:

Public Member Functions
	GraphTestcase ()
loco::Graph *	graph ()

Data Fields
loco::Pull *	pull_node = nullptr
loco::FeatureEncode *	encode_node = nullptr
loco::FeatureDecode *	decode_node = nullptr
loco::Push *	push_node = nullptr

Detailed Description

Definition at line 170 of file GraphTestcase.h.

Constructor & Destructor Documentation

GraphTestcase()

GraphTestcase< GraphCode::FeatureCodec >::GraphTestcase ( )

inline

Definition at line 173 of file GraphTestcase.h.

  {
    using namespace loco;
 
    Permutation<Domain::Feature> perm;
 
    perm[FeatureAxis::Count] = 0;
    perm[FeatureAxis::Height] = 1;
    perm[FeatureAxis::Width] = 2;
    perm[FeatureAxis::Depth] = 3;
 
    // Create a sample network
    _graph = make_graph();
 
    auto graph_builder = make_graph_builder(_graph.get());
 
    pull_node = graph_builder->push<InputLayer>()->name("input")->node();
    encode_node = graph_builder->push<FeatureEncodeLayer>()->perm(perm)->node();
    decode_node = graph_builder->push<FeatureDecodeLayer>()->perm(perm)->node();
    push_node = graph_builder->push<OutputLayer>()->name("output")->node();
  }

Member Function Documentation

graph()

loco::Graph * GraphTestcase< GraphCode::FeatureCodec >::graph ( )

inline

Definition at line 196 of file GraphTestcase.h.

{ return _graph.get(); }

Referenced by TopologicalSortHelper.TopologicalSortHelper::add_edge(), and TopologicalSortHelper.TopologicalSortHelper::sort_util().

Field Documentation

decode_node

loco::FeatureDecode* GraphTestcase< GraphCode::FeatureCodec >::decode_node = nullptr

Definition at line 200 of file GraphTestcase.h.

encode_node

loco::FeatureEncode* GraphTestcase< GraphCode::FeatureCodec >::encode_node = nullptr

Definition at line 199 of file GraphTestcase.h.

pull_node

loco::Pull* GraphTestcase< GraphCode::FeatureCodec >::pull_node = nullptr

Definition at line 198 of file GraphTestcase.h.

push_node

loco::Push* GraphTestcase< GraphCode::FeatureCodec >::push_node = nullptr

Definition at line 201 of file GraphTestcase.h.

---

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

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