|
std::ostream & | operator<< (std::ostream &os, const ComputationGraphBuilder::ComputableInfo &info) |
| This is to be used in logging only. More...
|
|
void | AddOutputToGraph (const ComputationRequest &request, const Nnet &nnet, ComputationGraph *graph) |
|
void | AddInputToGraph (const ComputationRequest &request, const Nnet &nnet, ComputationGraph *graph) |
|
static void | ComputeDependenciesSubset (const ComputationGraph &graph, const std::vector< int32 > &cindex_id_to_segment_and_epoch, std::vector< std::vector< int32 > > *dependencies_subset) |
| This function outputs to dependencies_subset[c], for each cindex_id c, the subset of elements d of graph.dependencies[c] such that cindex_id_to_segment_and_epoch[d] == cindex_id_to_segment_and_epoch[c]. More...
|
|
static void | ComputeEpochInfo (const Nnet &nnet, const ComputationGraph &graph, std::vector< int32 > *cindex_id_to_segment_and_epoch, std::vector< std::vector< std::vector< int32 > > > *epochs_per_segment, std::vector< bool > *epoch_is_trivial) |
| This function computes certain information about "epochs" of cindex_ids. More...
|
|
void | ComputeComputationGraph (const Nnet &nnet, const ComputationRequest &request, ComputationGraph *graph) |
|
static int32 | SumVectorSizes (const std::vector< std::vector< int32 > > &vec) |
|
static int32 | SumVectorSizes (const std::vector< std::vector< std::vector< int32 > > > &vec) |
|
static void | ComputeComputationPhasesForEpoch (const Nnet &nnet, const ComputationGraph &graph, const std::vector< int32 > &this_epoch, const std::vector< std::vector< int32 > > &dependencies_subset, const std::vector< std::vector< int32 > > &depend_on_subset, bool epoch_is_trivial, std::vector< int32 > *phase_indexes, std::vector< std::vector< int32 > > *phases) |
|
void | ComputeComputationPhases (const Nnet &nnet, const ComputationGraph &computation_graph, std::vector< std::vector< std::vector< int32 > > > *phases_per_segment) |
| This function divides a computation into 'phases', where a 'phase' is a collection of cindexes which can (as far as the computation graph is concerned) all be computed at the same time, and depend only on cindexes previously computed in earlier phases. More...
|
|