Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Xtensa LSTM: #2150

Merged
merged 17 commits into from
Jan 2, 2024
Merged

Xtensa LSTM: #2150

Show file tree
Hide file tree
Changes from 4 commits
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
d678c14
Xtensa LSTM:
cad-audio Jul 31, 2023
378aa87
Merge branch 'main' into xtensa_lstm_pr
advaitjain Aug 15, 2023
44adbe5
Optimized TFLM LSTM integration code for HiFi5/4 for 8x8 and 16x8 LSTM
cad-audio Sep 4, 2023
63947fb
Disbled the unit test cases under lstm_eval_test.cc for the XTENSA ta…
cad-audio Sep 4, 2023
02480ed
Fix style and remove hifi code from VP6 build
rascani Sep 20, 2023
d4b5703
Merge branch 'main' into xtensa_lstm_pr
advaitjain Sep 21, 2023
5dc5848
Updated the xtensa_download.sh file to download the latest cadence hi…
cad-audio Sep 22, 2023
9d03595
Merge branch 'main' into xtensa_lstm_pr
rascani Sep 26, 2023
8525ef3
Fix transpose_conv formatting
rascani Sep 26, 2023
8f67684
Updated the xa_nn_memmove_16 calling function
cad-audio Dec 4, 2023
7d62f18
Fix formatting
rascani Dec 5, 2023
3274767
Merge remote-tracking branch 'upstream_gh/main' into xtensa_lstm_pr
rascani Dec 5, 2023
0f210d0
Add HIFI3 defines alongside HIFI4 uses
rascani Dec 5, 2023
5a1d185
Fix typo
rascani Dec 5, 2023
9d2358b
Add DISCARD_FUN_FOR_NONVOID_RETURN to ndsp-hifi5
rascani Dec 5, 2023
ce19dd7
Added patch for conv 2d kernel for hifi4.
cad-audio Dec 21, 2023
d4c0e14
Merge branch 'main' into xtensa_lstm_pr
rascani Jan 2, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions tensorflow/lite/micro/kernels/lstm_eval_test.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ namespace {
// Test Settings
constexpr float kTestFloatTolerance = 1e-6f;
} // namespace
#endif // !defined(XTENSA)
#endif // !defined(XTENSA)

TF_LITE_MICRO_TESTS_BEGIN
// TODO(b/230666079) enable below tests for xtensa when the xtensa
Expand Down Expand Up @@ -454,6 +454,6 @@ TF_LITE_MICRO_TEST(TestLSTMEvalInt16) {
cell_state_tolerance,
int16_node_contents);
}
#endif // !defined(XTENSA)

#endif // !defined(XTENSA)
TF_LITE_MICRO_TESTS_END
2 changes: 0 additions & 2 deletions tensorflow/lite/micro/kernels/unidirectional_sequence_lstm_test.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -150,7 +150,6 @@ void TestUnidirectionalLSTMFloat(
TF_LITE_MICRO_TESTS_BEGIN
// TODO(b/230666079) enable below tests for xtensa when the xtensa
// kernel is reconciled with reference kernel
#if !defined(XTENSA)
TF_LITE_MICRO_TEST(TestUnidirectionalLSTMFloat) {
const tflite::testing::LstmEvalCheckData<12, 4, 12> kernel_eval_data =
tflite::testing::Get2X2LstmEvalCheckData();
Expand Down Expand Up @@ -193,5 +192,4 @@ TF_LITE_MICRO_TEST(TestUnidirectionalLSTMInt16) {
kernel_eval_data, hidden_state_tolerance, cell_state_tolerance,
int16_node_contents);
}
#endif // !defined(XTENSA)
TF_LITE_MICRO_TESTS_END
1,569 changes: 424 additions & 1,145 deletions tensorflow/lite/micro/kernels/xtensa/lstm_eval.cc
View file
Open in desktop

Large diffs are not rendered by default.

998 changes: 806 additions & 192 deletions tensorflow/lite/micro/kernels/xtensa/lstm_eval.h
View file
Open in desktop

Large diffs are not rendered by default.

Loading