Blame view
src/cudadecoder/decodable-cumatrix.h
2.33 KB
8dcb6dfcb
first commit
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
// cudadecoder/decodable-cumatrix.h
/*
* Copyright (c) 2017, NVIDIA CORPORATION. All rights reserved.
* Authors: Hugo Braun, Justin Luitjens, Ryan Leary
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef KALDI_CUDA_DECODER_DECODABLE_CUMATRIX_H_
#define KALDI_CUDA_DECODER_DECODABLE_CUMATRIX_H_
#include "cudadecoder/cuda-decodable-itf.h"
#include "cudamatrix/cu-matrix.h"
#include "decoder/decodable-matrix.h"
namespace kaldi {
namespace cuda_decoder {
/**
Cuda Decodable matrix. Takes transition model and posteriors and provides
an interface similar to the Decodable Interface
*/
class DecodableCuMatrixMapped : public CudaDecodableInterface {
public:
// This constructor creates an object that will not delete "likes" when done.
// the frame_offset is the frame the row 0 of 'likes' corresponds to, would be
// greater than one if this is not the first chunk of likelihoods.
DecodableCuMatrixMapped(const TransitionModel &tm,
const CuMatrixBase<BaseFloat> &likes,
int32 frame_offset = 0);
virtual int32 NumFramesReady() const;
virtual bool IsLastFrame(int32 frame) const;
virtual BaseFloat LogLikelihood(int32 frame, int32 tid) {
KALDI_ASSERT(false);
return 0.0f; // never executed, compiler requests a return
};
// Note: these indices are 1-based.
virtual int32 NumIndices() const;
virtual ~DecodableCuMatrixMapped(){};
// returns cuda pointer to nnet3 output
virtual BaseFloat *GetLogLikelihoodsCudaPointer(int32 subsampled_frame);
private:
const TransitionModel &trans_model_; // for tid to pdf mapping
const CuMatrixBase<BaseFloat> *likes_;
int32 frame_offset_;
KALDI_DISALLOW_COPY_AND_ASSIGN(DecodableCuMatrixMapped);
};
} // end namespace cuda_decoder
} // end namespace kaldi.
#endif // KALDI_CUDA_DECODER_DECODABLE_CUMATRIX_H_
|