diff options
Diffstat (limited to 'testing/web-platform/tests/webnn/resources/test_data/softmax.json')
| -rw-r--r-- | testing/web-platform/tests/webnn/resources/test_data/softmax.json | 203 |
1 files changed, 203 insertions, 0 deletions
diff --git a/testing/web-platform/tests/webnn/resources/test_data/softmax.json b/testing/web-platform/tests/webnn/resources/test_data/softmax.json new file mode 100644 index 0000000000..ebb12114fc --- /dev/null +++ b/testing/web-platform/tests/webnn/resources/test_data/softmax.json @@ -0,0 +1,203 @@ +{ + "tests": [ + { + "name": "softmax float32 2D constant tensor all positive", + "inputs": { + "x": { + "shape": [4, 6], + "data": [ + 7.903734730203054, + 6.358251623091922, + 4.833756774572475, + 9.579165187926348, + 0.21071856783621445, + 4.554958961778823, + 7.150174113718282, + 8.330297576341172, + 1.535985791562966, + 6.633619497318013, + 1.45393689965033, + 0.21341864769940821, + 5.257819475802503, + 8.1921378632378, + 8.161730206898543, + 2.8744343334015565, + 8.950733525100087, + 6.111632812590415, + 1.6371468489476149, + 0.2762612074523485, + 5.028227187056009, + 3.898326029997037, + 2.8967114227178836, + 6.8894703393234575 + ], + "type": "float32", + "constant": true + } + }, + "expected": { + "name": "output", + "shape": [4, 6], + "data": [ + 0.15068615972995758, + 0.03212761878967285, + 0.006995180621743202, + 0.8048291206359863, + 0.00006871300138300285, + 0.005293202120810747, + 0.2057899534702301, + 0.6698001027107239, + 0.0007502624066546559, + 0.1227685883641243, + 0.0006911618984304368, + 0.00019990770670119673, + 0.012398251332342625, + 0.23319464921951294, + 0.22621041536331177, + 0.0011435872875154018, + 0.4979347288608551, + 0.029118351638317108, + 0.004253828432410955, + 0.001090824487619102, + 0.12633030116558075, + 0.040812913328409195, + 0.014990009367465973, + 0.8125221133232117 + ], + "type": "float32" + } + }, + { + "name": "softmax float32 2D tensor all positive", + "inputs": { + "x": { + "shape": [4, 6], + "data": [ + 7.903734730203054, + 6.358251623091922, + 4.833756774572475, + 9.579165187926348, + 0.21071856783621445, + 4.554958961778823, + 7.150174113718282, + 8.330297576341172, + 1.535985791562966, + 6.633619497318013, + 1.45393689965033, + 0.21341864769940821, + 5.257819475802503, + 8.1921378632378, + 8.161730206898543, + 2.8744343334015565, + 8.950733525100087, + 6.111632812590415, + 1.6371468489476149, + 0.2762612074523485, + 5.028227187056009, + 3.898326029997037, + 2.8967114227178836, + 6.8894703393234575 + ], + "type": "float32" + } + }, + "expected": { + "name": "output", + "shape": [4, 6], + "data": [ + 0.15068615972995758, + 0.03212761878967285, + 0.006995180621743202, + 0.8048291206359863, + 0.00006871300138300285, + 0.005293202120810747, + 0.2057899534702301, + 0.6698001027107239, + 0.0007502624066546559, + 0.1227685883641243, + 0.0006911618984304368, + 0.00019990770670119673, + 0.012398251332342625, + 0.23319464921951294, + 0.22621041536331177, + 0.0011435872875154018, + 0.4979347288608551, + 0.029118351638317108, + 0.004253828432410955, + 0.001090824487619102, + 0.12633030116558075, + 0.040812913328409195, + 0.014990009367465973, + 0.8125221133232117 + ], + "type": "float32" + } + }, + { + "name": "softmax float32 2D tensor all negative", + "inputs": { + "x": { + "shape": [4, 6], + "data": [ + -3.3118434934513763, + -3.338954812269648, + -3.4102789638525355, + -6.697192980301335, + -7.896223320202411, + -3.3081689006635866, + -3.2309720485303473, + -4.315771351061817, + -9.311088112018716, + -3.9236627720798856, + -3.7807213277512064, + -6.034926915403203, + -3.9196677453714273, + -2.223484249593408, + -9.326531526135806, + -1.4882492282868132, + -6.302842393071306, + -5.531475594479687, + -1.8421411696525603, + -4.994808274460977, + -9.527291818875955, + -4.985682906583053, + -8.421041200931548, + -6.235629579015209 + ], + "type": "float32" + } + }, + "expected": { + "name": "output", + "shape": [4, 6], + "data": [ + 0.2546302080154419, + 0.24781952798366547, + 0.2307596504688263, + 0.008623254485428333, + 0.002599793951958418, + 0.2555675804615021, + 0.40352678298950195, + 0.13637976348400116, + 0.0009232329903170466, + 0.20185552537441254, + 0.23287305235862732, + 0.024441635236144066, + 0.0551743283867836, + 0.3008708655834198, + 0.0002474947541486472, + 0.6276082992553711, + 0.0050902292132377625, + 0.011008745059370995, + 0.9090295433998108, + 0.0388500951230526, + 0.00041779119055718184, + 0.039206232875585556, + 0.0012629841221496463, + 0.011233373545110226 + ], + "type": "float32" + } + } + ] +}
\ No newline at end of file |