#include <nnet-computation-graph.h>

Collaboration diagram for CindexSet:

Public Member Functions
bool	operator() (const Cindex &cindex) const
	Parenthesis operator; returns true if this cindex exists in the set. More...

	CindexSet (const ComputationGraph &graph)
	with this constructor, represents the set of all Cindexes that exist in the graph. More...

	CindexSet (const ComputationGraph &graph, const std::vector< ComputationGraphBuilder::CindexInfo > &info, bool treat_unknown_as_computable)
	with this constructor, represents the set of all Cindexes that exist in the graph and which are computable. More...

Private Attributes
const ComputationGraph &	graph_

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

bool	treat_unknown_as_computable_

Detailed Description

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

Constructor & Destructor Documentation

◆ CindexSet() [1/2]

CindexSet ( const ComputationGraph & graph )

with this constructor, represents the set of all Cindexes that exist in the graph.

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

1466 :

1467 graph_(graph), info_(NULL) { }

kaldi::nnet3::CindexSet::info_

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

Definition: nnet-computation-graph.h:316

kaldi::nnet3::CindexSet::graph_

const ComputationGraph & graph_

Definition: nnet-computation-graph.h:315

◆ CindexSet() [2/2]

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

with this constructor, represents the set of all Cindexes that exist in the graph and which are computable.

If treat_unknown_as_computable is true then we consider kComputable, kUnknown and kWillNotCompute to be computable; else we consider just nodes that are kComputable to be computable.

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

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

Member Function Documentation

◆ operator()()

bool operator() ( const Cindex & cindex ) const

Parenthesis operator; returns true if this cindex exists in the set.

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

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

                                                        {
   int32 cindex_id = graph_.GetCindexId(cindex);
   if (cindex_id == -1) {
     return false;
   } else {
     if (info_ == NULL) {
       return true;
     } else {
       ComputationGraphBuilder::ComputableInfo
           c = (*info_)[cindex_id].computable;
       return (c == ComputationGraphBuilder::kComputable ||
               (c == ComputationGraphBuilder::kUnknown &&
                treat_unknown_as_computable_));
     }
   }
 }