Skip to content

Commit

Permalink
add WebNN in log
Browse files Browse the repository at this point in the history
  • Loading branch information
shiyi9801 committed Oct 24, 2024
1 parent 9f62135 commit 7f47bc2
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ bool GatherNDOpBuilder::IsOpSupportedImpl(const InitializedTensorSet& /* initial
const logging::Logger& logger) const {
NodeAttrHelper helper(node);
if (helper.Get("batch_dims", 0) != 0) {
LOGS(logger, VERBOSE) << "GatherND: batch_dims must be equal to 0";
LOGS(logger, VERBOSE) << "GatherND: WebNN only supports batch_dims 0 (default)";
return false;
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ bool ScatterNDOpBuilder::IsOpSupportedImpl(const InitializedTensorSet& /* initia
const logging::Logger& logger) const {
NodeAttrHelper helper(node);
if (helper.Get("reduction", "none") != "none") {
LOGS(logger, VERBOSE) << "ScatterND: the type of reduction must be none (default)";
LOGS(logger, VERBOSE) << "ScatterND: WebNN only supports reduction type none (default)";
return false;
}

Expand Down

0 comments on commit 7f47bc2

Please sign in to comment.