mirrors/kaldifeat
1
0
Fork 0
mirror of https://github.com/csukuangfj/kaldifeat.git synced 2025-08-10 10:32:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix android build (#40)

Browse Source
This commit is contained in:
Wei Kang 2022-07-04 16:28:34 +08:00 committed by GitHub
parent c6e643951f
commit cc1cd67c3e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
kaldifeat/csrc/feature-common-inl.h
View File

@ -55,10 +55,17 @@ torch::Tensor OfflineFeatureTpl<F>::ComputeFeatures(const torch::Tensor &wave,
int32_t padding = frame_opts.PaddedWindowSize() - strided_input.size(1); int32_t padding = frame_opts.PaddedWindowSize() - strided_input.size(1);
if (padding > 0) { if (padding > 0) {
#ifdef __ANDROID__
auto padding_value = torch::zeros(
{strided_input.size(0), padding},
torch::dtype(torch::kFloat).device(strided_input.device()));
strided_input = torch::cat({strided_input, padding_value}, 1);
#else
strided_input = torch::nn::functional::pad( strided_input = torch::nn::functional::pad(
strided_input, torch::nn::functional::PadFuncOptions({0, padding}) strided_input, torch::nn::functional::PadFuncOptions({0, padding})
.mode(torch::kConstant) .mode(torch::kConstant)
.value(0)); .value(0));
#endif
} }
return computer_.Compute(log_energy_pre_window, vtln_warp, strided_input); return computer_.Compute(log_energy_pre_window, vtln_warp, strided_input);