This is the complete list of members for NnetBatchComputer, including all inherited members.
AcceptTask(NnetInferenceTask *task, int32 max_minibatches_full=-1) | NnetBatchComputer | |
CheckAndFixConfigs() | NnetBatchComputer | private |
compiler_ | NnetBatchComputer | private |
Compute(bool allow_partial_minibatch) | NnetBatchComputer | |
FormatInputs(int32 minibatch_size, const std::vector< NnetInferenceTask *> &tasks, CuMatrix< BaseFloat > *input, CuMatrix< BaseFloat > *ivector) | NnetBatchComputer | private |
FormatOutputs(const CuMatrix< BaseFloat > &output, const std::vector< NnetInferenceTask *> &tasks) | NnetBatchComputer | private |
GetActualMinibatchSize(const ComputationGroupInfo &info) const | NnetBatchComputer | private |
GetComputation(const ComputationGroupInfo &info, int32 minibatch_size) | NnetBatchComputer | private |
GetComputationRequest(const NnetInferenceTask &task, int32 minibatch_size, ComputationRequest *request) | NnetBatchComputer | privatestatic |
GetHighestPriorityComputation(bool allow_partial_minibatch, int32 *minibatch_size, std::vector< NnetInferenceTask *> *tasks) | NnetBatchComputer | private |
GetHighestPriorityTasks(int32 num_tasks, ComputationGroupInfo *info, std::vector< NnetInferenceTask *> *tasks) | NnetBatchComputer | private |
GetMinibatchSize(const ComputationGroupInfo &info) const | NnetBatchComputer | inlineprivate |
GetOptions() | NnetBatchComputer | inline |
GetPriority(bool allow_partial_minibatch, const ComputationGroupInfo &info) const | NnetBatchComputer | inlineprivate |
input_dim_ | NnetBatchComputer | private |
ivector_dim_ | NnetBatchComputer | private |
KALDI_DISALLOW_COPY_AND_ASSIGN(NnetBatchComputer) | NnetBatchComputer | private |
log_priors_ | NnetBatchComputer | private |
MapType typedef | NnetBatchComputer | private |
mutex_ | NnetBatchComputer | private |
nnet_ | NnetBatchComputer | private |
nnet_left_context_ | NnetBatchComputer | private |
nnet_right_context_ | NnetBatchComputer | private |
NnetBatchComputer(const NnetBatchComputerOptions &opts, const Nnet &nnet, const VectorBase< BaseFloat > &priors) | NnetBatchComputer | |
no_more_than_n_minibatches_full_ | NnetBatchComputer | private |
num_full_minibatches_ | NnetBatchComputer | private |
NumFullPendingMinibatches() const | NnetBatchComputer | inline |
opts_ | NnetBatchComputer | private |
output_dim_ | NnetBatchComputer | private |
PrintMinibatchStats() | NnetBatchComputer | private |
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_ | NnetBatchComputer | private |
~NnetBatchComputer() | NnetBatchComputer |