An abstract representation of a set of Indexes. More...

#include <nnet-computation-graph.h>

Collaboration diagram for IndexSet:

Public Member Functions
bool	operator() (const Index &index) const
	Returns true if this Index exists in the set. More...

	IndexSet (const ComputationGraph &graph, const std::vector< ComputationGraphBuilder::CindexInfo > &info, int32 node_id, bool treat_unknown_as_computable)
	This constructor creates the set of all Indexes x such that a Cindex (node_id, x) which is computable exists in this graph. More...

Private Attributes
const ComputationGraph &	graph_

const std::vector< ComputationGraphBuilder::CindexInfo > &	info_

int32	node_id_

bool	treat_unknown_as_computable_

Detailed Description

An abstract representation of a set of Indexes.

Definition at line 322 of file nnet-computation-graph.h.

Constructor & Destructor Documentation

◆ IndexSet()

IndexSet	(	const ComputationGraph &	graph,
		const std::vector< ComputationGraphBuilder::CindexInfo > &	info,
		int32	node_id,
		bool	treat_unknown_as_computable
	)

This constructor creates the set of all Indexes x such that a Cindex (node_id, x) which is computable exists in this graph.

If treat_unknown_as_computable is true then we consider kComputable and kUnknown to be computable, else we consider just nodes that are kComputable to be computable. The `info` input is only needed for its `computable` member.

Definition at line 1493 of file nnet-computation-graph.cc.

                                                     :
     graph_(graph), info_(info), node_id_(node_id),
     treat_unknown_as_computable_(treat_unknown_as_computable) { }

Member Function Documentation

◆ operator()()

bool operator() ( const Index & index ) const

Returns true if this Index exists in the set.

Definition at line 1500 of file nnet-computation-graph.cc.

References ComputationGraph::GetCindexId(), IndexSet::graph_, IndexSet::info_, ComputationGraphBuilder::kComputable, ComputationGraphBuilder::kUnknown, IndexSet::node_id_, and IndexSet::treat_unknown_as_computable_.

                                                     {
   int32 cindex_id = graph_.GetCindexId(Cindex(node_id_, index));
   if (cindex_id == -1) {
     return false;
   } else {
     ComputationGraphBuilder::ComputableInfo c = info_[cindex_id].computable;
     if (treat_unknown_as_computable_)
       return (c == ComputationGraphBuilder::kComputable ||
               c == ComputationGraphBuilder::kUnknown);
     else
       return (c == ComputationGraphBuilder::kComputable);
   }
 }