saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-28 20:11:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[WebNN EP] Change axis to axes for argMax/argMin (#18838)

Browse source 
In the latest spec, the axes option of WebNN's argMax and argMin
requires the use of a sequence long type. Replace axis option (long
type) with axes (sequence long type) for argMax and argMin.
This commit is contained in:
Peishen Yan 2023-12-16 00:57:07 +08:00 committed by GitHub
parent d795fc636c
commit d111eed726
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/providers/webnn/builders/impl/argmax_min_op_builder.cc
View file

@ -41,9 +41,11 @@ Status ArgMaxMinOpBuilder::AddToModelBuilderImpl(ModelBuilder& model_builder,
const auto select_last_index = helper.Get("select_last_index", 0);
axis = HandleNegativeAxis(axis, input_rank);
emscripten::val axes = emscripten::val::array();
axes.call<void>("push", static_cast<uint32_t>(axis));
emscripten::val options = emscripten::val::object();
options.set("axis", static_cast<int32_t>(axis));
options.set("axes", axes);
options.set("keepDimensions", keep_dims == 1);
options.set("selectLastIndex", select_last_index == 1);
emscripten::val output = emscripten::val::object();