Skip to content
Commit 835d8e10 authored by Fredrik Svedberg's avatar Fredrik Svedberg Committed by patrik.gustavsson
Browse files

[MLBEDSW-2928] Add batching to softmax 



Added batching to softmax by reshaping the input.

Signed-off-by: Fredrik Svedberg's avatarFredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I0b516f9bf2410fb86372b229beba4a7280c498cc
parent e5cf95b8
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment