Home
last modified time | relevance | path

Searched refs:_test_fully_connected (Results 1 – 3 of 3) sorted by relevance

/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/frontend/tflite/
H A Dtest_forward.py977 def _test_fully_connected(tensor_in_sizes, filter_in_sizes, bias_in_size=None): function
1015 _test_fully_connected([1, 1, 1, 150], [150, 100])
1016 _test_fully_connected([1, 1, 1, 150], [150, 100], [100])
1017 _test_fully_connected([5, 1, 1, 150], [150, 100])
1018 _test_fully_connected([5, 1, 1, 150], [150, 100], [100])
/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/frontend/tflite/
H A Dtest_forward.py977 def _test_fully_connected(tensor_in_sizes, filter_in_sizes, bias_in_size=None): function
1015 _test_fully_connected([1, 1, 1, 150], [150, 100])
1016 _test_fully_connected([1, 1, 1, 150], [150, 100], [100])
1017 _test_fully_connected([5, 1, 1, 150], [150, 100])
1018 _test_fully_connected([5, 1, 1, 150], [150, 100], [100])
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/frontend/tflite/
H A Dtest_forward.py3182 def _test_fully_connected(tensor_in_sizes, const_input, filter_in_sizes, bias_in_size=None): function
3226 _test_fully_connected([1, 1, 1, 150], const_input, [150, 100])
3227 _test_fully_connected([1, 1, 1, 150], const_input, [150, 100], [100])
3228 _test_fully_connected([5, 1, 1, 150], const_input, [150, 100])
3229 _test_fully_connected([5, 1, 1, 150], const_input, [150, 100], [100])