Public Attributes | List of all members
ConvolutionComputation::ConvolutionStep Struct Reference

#include <convolution.h>

Collaboration diagram for ConvolutionComputation::ConvolutionStep:
Collaboration graph
[legend]

Public Attributes

int32 input_time_shift
 
int32 params_start_col
 
std::vector< int32 > height_map
 
CuArray< int32 > columns
 
std::vector< CuArray< int32 > > backward_columns
 
bool columns_are_contiguous
 
int32 first_column
 

Detailed Description

Definition at line 290 of file convolution.h.

Member Data Documentation

◆ backward_columns

std::vector<CuArray<int32> > backward_columns

Definition at line 332 of file convolution.h.

Referenced by ConvolutionComputation::Check(), ConvolutionComputation::ComputeDerived(), and kaldi::nnet3::time_height_convolution::ConvolveBackwardDataInternal().

◆ columns

CuArray<int32> columns

Definition at line 323 of file convolution.h.

Referenced by ConvolutionComputation::Check(), ConvolutionComputation::ComputeDerived(), kaldi::nnet3::time_height_convolution::ConvolveBackwardDataInternal(), kaldi::nnet3::time_height_convolution::ConvolveBackwardParamsInternal(), and kaldi::nnet3::time_height_convolution::ConvolveForwardInternal().

◆ columns_are_contiguous

bool columns_are_contiguous

Definition at line 337 of file convolution.h.

Referenced by ConvolutionComputation::Check(), ConvolutionComputation::ComputeDerived(), kaldi::nnet3::time_height_convolution::ConvolveBackwardDataInternal(), kaldi::nnet3::time_height_convolution::ConvolveBackwardParamsInternal(), and kaldi::nnet3::time_height_convolution::ConvolveForwardInternal().

◆ first_column

int32 first_column

Definition at line 341 of file convolution.h.

Referenced by ConvolutionComputation::Check(), ConvolutionComputation::ComputeDerived(), kaldi::nnet3::time_height_convolution::ConvolveBackwardDataInternal(), kaldi::nnet3::time_height_convolution::ConvolveBackwardParamsInternal(), and kaldi::nnet3::time_height_convolution::ConvolveForwardInternal().

◆ height_map

std::vector<int32> height_map

Definition at line 313 of file convolution.h.

Referenced by ConvolutionComputation::Check(), ConvolutionComputation::ComputeDerived(), kaldi::nnet3::time_height_convolution::ComputeTempMatrixSize(), kaldi::nnet3::time_height_convolution::MakeComputation(), ConvolutionComputation::Read(), kaldi::nnet3::time_height_convolution::UnPadModelHeight(), and ConvolutionComputation::Write().

◆ input_time_shift

int32 input_time_shift

Definition at line 296 of file convolution.h.

Referenced by ConvolutionComputation::Check(), kaldi::nnet3::time_height_convolution::ConvolveBackwardDataInternal(), kaldi::nnet3::time_height_convolution::ConvolveBackwardParamsInternal(), kaldi::nnet3::time_height_convolution::ConvolveForwardInternal(), kaldi::nnet3::time_height_convolution::MakeComputation(), ConvolutionComputation::Read(), and ConvolutionComputation::Write().

◆ params_start_col

int32 params_start_col

Definition at line 301 of file convolution.h.

Referenced by ConvolutionComputation::Check(), kaldi::nnet3::time_height_convolution::ConvolveBackwardDataInternal(), kaldi::nnet3::time_height_convolution::ConvolveBackwardParamsInternal(), kaldi::nnet3::time_height_convolution::ConvolveForwardInternal(), kaldi::nnet3::time_height_convolution::MakeComputation(), ConvolutionComputation::Read(), and ConvolutionComputation::Write().

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