Home
last modified time | relevance | path

Searched refs:softmax (Results 1 – 7 of 7) sorted by relevance

/OK3568_Linux_fs/external/rknn-toolkit2/examples/pytorch/resnet18/
H A Dtest.py38 def softmax(x): function
98 show_outputs(softmax(np.array(outputs[0][0])))
/OK3568_Linux_fs/external/rknn-toolkit2/examples/pytorch/resnet18_qat/
H A Dtest.py31 def softmax(x): function
100 show_outputs(softmax(np.array(outputs[0][0])))
/OK3568_Linux_fs/external/rknn-toolkit2/examples/functions/hybrid_quant/
H A Dssd_post_process.py47 def softmax(x): function
79 conf_data[i] = softmax(conf_data[i])
/OK3568_Linux_fs/external/rknn-toolkit2/doc/
H A Dchangelog-1.5.0.txt209 6. 优化dataconvert/gather/transpose/mul/maxpool/sigmoid/pad/conv/relu/softmax支持
233 8. 增加swish/dataconvert/softmax/lstm/layernorm相关优化
H A DRKNNToolKit2_OP_Support-1.5.0.md346 | aten::softmax | same as onnx Softmax |
491 | softmax | …
/OK3568_Linux_fs/docs/en/Common/NPU/rknn-toolkit2/
H A DRKNNToolKit2_OP_Support-1.4.0.md439 | aten::softmax | same as onnx Softmax …
532 | softmax | …
/OK3568_Linux_fs/docs/cn/Common/NPU/rknn-toolkit2/
H A DRKNNToolKit2_OP_Support-1.4.0.md439 | aten::softmax | same as onnx Softmax …
532 | softmax | …