Home
last modified time | relevance | path

Searched refs:input_size_list (Results 1 – 10 of 10) sorted by relevance

/OK3568_Linux_fs/external/rknn-toolkit2/examples/pytorch/resnet18/
H A Dtest.py48 input_size_list = [[1, 3, 224, 224]] variable
60 ret = rknn.load_pytorch(model=model, input_size_list=input_size_list)
/OK3568_Linux_fs/external/rknn-toolkit2/examples/pytorch/resnet18_qat/
H A Dtest.py50 input_size_list = [[1, 3, 224, 224]] variable
62 ret = rknn.load_pytorch(model=model, input_size_list=input_size_list)
/OK3568_Linux_fs/docs/cn/Common/NPU/rknn-toolkit2/
H A DRKNNToolKit2_API_Difference_With_Toolkit1-1.4.0.md78 input_size_list,
89 input_size_list,
126 input_size_list=None,
133 input_size_list)
138 load_mxnet(symbol, params, input_size_list=None)
/OK3568_Linux_fs/external/rknn-toolkit2/doc/
H A DRKNNToolKit2_API_Difference_With_Toolkit1-1.5.0.md81 input_size_list,
92 input_size_list,
129 input_size_list=None,
136 input_size_list)
141 load_mxnet(symbol, params, input_size_list=None)
/OK3568_Linux_fs/docs/en/Common/NPU/rknn-toolkit2/
H A DRKNNToolKit2_API_Difference_With_Toolkit1-1.4.0.md78 input_size_list,
89 input_size_list,
126 input_size_list=None,
133 input_size_list)
138 load_mxnet(symbol, params, input_size_list=None)
/OK3568_Linux_fs/external/rknn-toolkit2/examples/functions/hybrid_quant/
H A Dstep1.py20 input_size_list=[[1,300,300,3]])
/OK3568_Linux_fs/external/rknn-toolkit2/examples/functions/multi_input_test/
H A Dtest.py21 … input_size_list=[[1, 128, 128, 3], [1, 128, 128, 3], [1, 128, 128, 3], [1, 128, 128, 1]])
/OK3568_Linux_fs/external/rknn-toolkit2/examples/functions/mmse/
H A Dtest.py39 input_size_list=[[1, 224, 224, 3]],
/OK3568_Linux_fs/external/rknn-toolkit2/examples/tensorflow/inception_v3_qat/
H A Dtest.py115 input_size_list=[[1, INPUT_SIZE, INPUT_SIZE, 3]])
/OK3568_Linux_fs/external/rknn-toolkit2/examples/tensorflow/ssd_mobilenet_v1/
H A Dtest.py73 input_size_list=[[1, INPUT_SIZE, INPUT_SIZE, 3]])