NnetBatchComputer Member List

This is the complete list of members for NnetBatchComputer, including all inherited members.

AcceptTask(NnetInferenceTask *task, int32 max_minibatches_full=-1)NnetBatchComputer
CheckAndFixConfigs()NnetBatchComputerprivate
compiler_NnetBatchComputerprivate
Compute(bool allow_partial_minibatch)NnetBatchComputer
FormatInputs(int32 minibatch_size, const std::vector< NnetInferenceTask *> &tasks, CuMatrix< BaseFloat > *input, CuMatrix< BaseFloat > *ivector)NnetBatchComputerprivate
FormatOutputs(const CuMatrix< BaseFloat > &output, const std::vector< NnetInferenceTask *> &tasks)NnetBatchComputerprivate
GetActualMinibatchSize(const ComputationGroupInfo &info) constNnetBatchComputerprivate
GetComputation(const ComputationGroupInfo &info, int32 minibatch_size)NnetBatchComputerprivate
GetComputationRequest(const NnetInferenceTask &task, int32 minibatch_size, ComputationRequest *request)NnetBatchComputerprivatestatic
GetHighestPriorityComputation(bool allow_partial_minibatch, int32 *minibatch_size, std::vector< NnetInferenceTask *> *tasks)NnetBatchComputerprivate
GetHighestPriorityTasks(int32 num_tasks, ComputationGroupInfo *info, std::vector< NnetInferenceTask *> *tasks)NnetBatchComputerprivate
GetMinibatchSize(const ComputationGroupInfo &info) constNnetBatchComputerinlineprivate
GetOptions()NnetBatchComputerinline
GetPriority(bool allow_partial_minibatch, const ComputationGroupInfo &info) constNnetBatchComputerinlineprivate
input_dim_NnetBatchComputerprivate
ivector_dim_NnetBatchComputerprivate
KALDI_DISALLOW_COPY_AND_ASSIGN(NnetBatchComputer)NnetBatchComputerprivate
log_priors_NnetBatchComputerprivate
MapType typedefNnetBatchComputerprivate
mutex_NnetBatchComputerprivate
nnet_NnetBatchComputerprivate
nnet_left_context_NnetBatchComputerprivate
nnet_right_context_NnetBatchComputerprivate
NnetBatchComputer(const NnetBatchComputerOptions &opts, const Nnet &nnet, const VectorBase< BaseFloat > &priors)NnetBatchComputer
no_more_than_n_minibatches_full_NnetBatchComputerprivate
num_full_minibatches_NnetBatchComputerprivate
NumFullPendingMinibatches() constNnetBatchComputerinline
opts_NnetBatchComputerprivate
output_dim_NnetBatchComputerprivate
PrintMinibatchStats()NnetBatchComputerprivate
SplitUtteranceIntoTasks(bool output_to_cpu, const Matrix< BaseFloat > &input, const Vector< BaseFloat > *ivector, const Matrix< BaseFloat > *online_ivectors, int32 online_ivector_period, std::vector< NnetInferenceTask > *tasks)NnetBatchComputer
SplitUtteranceIntoTasks(bool output_to_cpu, const CuMatrix< BaseFloat > &input, const CuVector< BaseFloat > *ivector, const CuMatrix< BaseFloat > *online_ivectors, int32 online_ivector_period, std::vector< NnetInferenceTask > *tasks)NnetBatchComputer
tasks_NnetBatchComputerprivate
~NnetBatchComputer()NnetBatchComputer