Skip to content

Commit 1226483

Browse files
committed
Added mem_stride to mli_tensor initializations
1 parent b795e49 commit 1226483

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

examples/example_kws_speech/Makefile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ BUILD_DIR ?= ./obj
3535
OUT_NAME ?= example_kws_stream
3636
ifeq ($(TOOLCHAIN),mwdt)
3737
# MWDT specific options
38-
CFLAGS = -Hnocopyr -Hpurge -Hheap=4K -Hstack=4K -Hdsplib -Hfxapi -e_start -Bgrouplib -Hldopt=-q -O3 -Hsdata0 -Xdsp_ctrl=postshift,guard,convergent -Hdense_prologue
38+
CFLAGS = -Hnocopyr -Hpurge -Hheap=4K -Hstack=4K -Hdsplib -Hfxapi -e_start -Bgrouplib -Hldopt=-q -Hsdata0 -Xdsp_ctrl=postshift,guard,convergent -Hdense_prologue
3939

4040
ifneq ($(RT_LIB),)
4141
CFLAGS += -Hlib=$(RT_LIB)

examples/example_kws_speech/kws/dsconv_lstm_nn/dsconv_lstm_nn_model_impl.cc

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -254,7 +254,7 @@ kws_status kws_dsconv_lstm_nn::audio_features_extract(const sample_t *in_frame,
254254
y_mem->nn.fex_phase.ir_buf_y, x_mem->nn.fex_phase.ir_buf_x);
255255
if (fbanks_num == kFbankNumBins) {
256256
mli_tensor out_fx_features = {
257-
(void *)out_features, kFbankNumBins * sizeof(out_features[0]), {0}, 0,
257+
(void *)out_features, kFbankNumBins * sizeof(out_features[0]), {0}, {0}, 0,
258258
MLI_EL_FX_8, {kFbankFraqBits}
259259
};
260260
if (MLI_STATUS_OK !=
@@ -280,7 +280,7 @@ kws_status kws_dsconv_lstm_nn::nn_inference(const int8_t *in_features, float *ou
280280
{
281281
mli_tensor input = {
282282
(void *)in_features, kFeatureVectorsForInference * kFbankNumBins * sizeof(in_features[0]),
283-
{1, kFeatureVectorsForInference, kFbankNumBins}, 3,
283+
{0}, {1, kFeatureVectorsForInference, kFbankNumBins}, 3,
284284
MLI_EL_FX_8, {kFbankFraqBits}
285285
};
286286

@@ -343,12 +343,12 @@ kws_status kws_dsconv_lstm_nn::nn_inference(const int8_t *in_features, float *ou
343343
// LAYER 5
344344
// init structures for LSTM layer
345345
const uint32_t lstm_cell_size = m->L5_lstm_bias.shape[1];
346-
mli_tensor lstm_prev_out = {ir_X.data, ir_X.capacity, {lstm_cell_size}, 1, MLI_EL_FX_16, {7}};
346+
mli_tensor lstm_prev_out = { ir_X.data, ir_X.capacity, {0}, {lstm_cell_size}, 1, MLI_EL_FX_16, {7} };
347347
mli_tensor lstm_ir = { (void *)x_mem->nn.rnn_phase.lstm_ir_data, sizeof(x_mem->nn.rnn_phase.lstm_ir_data) };
348348
const mli_rnn_cell_cfg lstm_cfg = {m->lstm_mode, m->lstm_act, &lstm_ir};
349349
mli_tensor lstm_cell = {
350350
(void *)y_mem->nn.rnn_phase.lstm_cell_data, sizeof(y_mem->nn.rnn_phase.lstm_cell_data),
351-
{lstm_cell_size}, 1,
351+
{0}, {lstm_cell_size}, 1,
352352
MLI_EL_FX_16, {m->L5_lstm_cell_fraq}
353353
};
354354

0 commit comments

Comments
 (0)