Public Member Functions | Public Attributes | List of all members
NnetInferenceTask Struct Reference

class NnetInferenceTask represents a chunk of an utterance that is requested to be computed. More...

#include <nnet-batch-compute.h>

Collaboration diagram for NnetInferenceTask:
Collaboration graph
[legend]

Public Member Functions

 NnetInferenceTask (const NnetInferenceTask &other)
 
 NnetInferenceTask ()
 

Public Attributes

CuMatrix< BaseFloat > input
 
int32 first_input_t
 
int32 output_t_stride
 
int32 num_output_frames
 
int32 num_initial_unused_output_frames
 
int32 num_used_output_frames
 
int32 first_used_output_frame_index
 
bool is_edge
 
bool is_irregular
 
CuVector< BaseFloat > ivector
 
double priority
 
Semaphore semaphore
 
bool output_to_cpu
 
Matrix< BaseFloat > output_cpu
 
CuMatrix< BaseFloat > output
 

Detailed Description

class NnetInferenceTask represents a chunk of an utterance that is requested to be computed.

This will be given to NnetBatchComputer, which will aggregate the tasks and complete them.

Definition at line 50 of file nnet-batch-compute.h.

Constructor & Destructor Documentation

◆ NnetInferenceTask() [1/2]

NnetInferenceTask ( const NnetInferenceTask &  other)
inline

Definition at line 53 of file nnet-batch-compute.h.

References KALDI_ERR.

53  {
54  KALDI_ERR << "NnetInferenceTask was not designed to be copied.";
55  }
KALDI_ERR
#define KALDI_ERR
Definition: kaldi-error.h:147

◆ NnetInferenceTask() [2/2]

NnetInferenceTask ( )
inline

Definition at line 56 of file nnet-batch-compute.h.

56 { }

Member Data Documentation

◆ first_input_t

int32 first_input_t

Definition at line 70 of file nnet-batch-compute.h.

Referenced by NnetBatchComputer::GetComputationRequest(), NnetBatchComputer::ComputationGroupKey::operator==(), and kaldi::nnet3::utterance_splitting::SplitInputToTasks().

◆ first_used_output_frame_index

int32 first_used_output_frame_index

Definition at line 102 of file nnet-batch-compute.h.

Referenced by kaldi::nnet3::utterance_splitting::AddOnlineIvectorsToTasks(), kaldi::nnet3::utterance_splitting::GetOutputFrameInfoForTasks(), kaldi::nnet3::MergeTaskOutput(), and kaldi::nnet3::utterance_splitting::SplitInputToTasks().

◆ input

CuMatrix<BaseFloat> input

Definition at line 63 of file nnet-batch-compute.h.

Referenced by NnetBatchComputer::GetComputationRequest(), NnetBatchComputer::NumFullPendingMinibatches(), and kaldi::nnet3::utterance_splitting::SplitInputToTasks().

◆ is_edge

bool is_edge

Definition at line 108 of file nnet-batch-compute.h.

Referenced by NnetBatchComputer::GetMinibatchSize(), and kaldi::nnet3::utterance_splitting::SplitInputToTasks().

◆ is_irregular

bool is_irregular

Definition at line 113 of file nnet-batch-compute.h.

Referenced by NnetBatchComputer::GetMinibatchSize(), and kaldi::nnet3::utterance_splitting::GetOutputFrameInfoForTasks().

◆ ivector

CuVector<BaseFloat> ivector

Definition at line 116 of file nnet-batch-compute.h.

Referenced by kaldi::nnet3::utterance_splitting::AddOnlineIvectorsToTasks(), NnetBatchComputer::GetComputationRequest(), and NnetBatchComputer::NumFullPendingMinibatches().

◆ num_initial_unused_output_frames

int32 num_initial_unused_output_frames

Definition at line 85 of file nnet-batch-compute.h.

Referenced by kaldi::nnet3::utterance_splitting::AddOnlineIvectorsToTasks(), NnetBatchDecoder::Decode(), NnetBatchComputer::FormatOutputs(), kaldi::nnet3::utterance_splitting::GetOutputFrameInfoForTasks(), kaldi::nnet3::MergeTaskOutput(), and kaldi::nnet3::utterance_splitting::SplitInputToTasks().

◆ num_output_frames

int32 num_output_frames

Definition at line 78 of file nnet-batch-compute.h.

Referenced by kaldi::nnet3::utterance_splitting::AddOnlineIvectorsToTasks(), NnetBatchComputer::GetComputationRequest(), kaldi::nnet3::utterance_splitting::GetOutputFrameInfoForTasks(), NnetBatchComputer::ComputationGroupKey::operator==(), and kaldi::nnet3::utterance_splitting::SplitInputToTasks().

◆ num_used_output_frames

int32 num_used_output_frames

Definition at line 90 of file nnet-batch-compute.h.

Referenced by NnetBatchDecoder::Decode(), NnetBatchComputer::FormatOutputs(), kaldi::nnet3::utterance_splitting::GetOutputFrameInfoForTasks(), and kaldi::nnet3::MergeTaskOutput().

◆ output

CuMatrix<BaseFloat> output

Definition at line 140 of file nnet-batch-compute.h.

Referenced by NnetBatchDecoder::Decode(), NnetBatchComputer::FormatOutputs(), kaldi::nnet3::MergeTaskOutput(), and NnetBatchComputerOptions::Register().

◆ output_cpu

Matrix<BaseFloat> output_cpu

Definition at line 137 of file nnet-batch-compute.h.

Referenced by NnetBatchDecoder::Decode(), NnetBatchComputer::FormatOutputs(), and kaldi::nnet3::MergeTaskOutput().

◆ output_t_stride

int32 output_t_stride

Definition at line 74 of file nnet-batch-compute.h.

Referenced by NnetBatchComputer::GetComputationRequest().

◆ output_to_cpu

bool output_to_cpu

Definition at line 130 of file nnet-batch-compute.h.

Referenced by NnetBatchComputer::FormatOutputs(), kaldi::nnet3::MergeTaskOutput(), and NnetBatchComputer::NumFullPendingMinibatches().

◆ priority

double priority

Definition at line 120 of file nnet-batch-compute.h.

Referenced by NnetBatchDecoder::ComputeFunc(), and NnetBatchDecoder::Decode().

◆ semaphore

Semaphore semaphore

Definition at line 125 of file nnet-batch-compute.h.

Referenced by NnetBatchDecoder::Decode().

The documentation for this struct was generated from the following file: