Searched refs:mx_rhs (Results 1 – 2 of 2) sorted by relevance
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
H A D | test_sparse_operator.py | 1910 def check_broadcast_add(mx_lhs, mx_rhs, np_lhs, np_rhs, dtype): argument 1912 def check_broadcast_sub(mx_lhs, mx_rhs, np_lhs, np_rhs, dtype): argument 1927 np_rhs = mx_rhs.asnumpy() 1928 check_broadcast_add(mx_lhs, mx_rhs, np_lhs, np_rhs, np.float32) 1929 check_broadcast_sub(mx_lhs, mx_rhs, np_lhs, np_rhs, np.float32) 1930 check_broadcast_add(mx_rhs, mx_lhs, np_rhs, np_lhs, np.float32) 1931 check_broadcast_sub(mx_rhs, mx_lhs, np_rhs, np_lhs, np.float32) 1935 def check_broadcast_mul(mx_lhs, mx_rhs, np_lhs, np_rhs, dtype): argument 1937 def check_broadcast_div(mx_lhs, mx_rhs, np_lhs, np_rhs, dtype): argument 1952 np_rhs = mx_rhs.asnumpy() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
H A D | test_sparse_operator.py | 1910 def check_broadcast_add(mx_lhs, mx_rhs, np_lhs, np_rhs, dtype): argument 1912 def check_broadcast_sub(mx_lhs, mx_rhs, np_lhs, np_rhs, dtype): argument 1927 np_rhs = mx_rhs.asnumpy() 1928 check_broadcast_add(mx_lhs, mx_rhs, np_lhs, np_rhs, np.float32) 1929 check_broadcast_sub(mx_lhs, mx_rhs, np_lhs, np_rhs, np.float32) 1930 check_broadcast_add(mx_rhs, mx_lhs, np_rhs, np_lhs, np.float32) 1931 check_broadcast_sub(mx_rhs, mx_lhs, np_rhs, np_lhs, np.float32) 1935 def check_broadcast_mul(mx_lhs, mx_rhs, np_lhs, np_rhs, dtype): argument 1937 def check_broadcast_div(mx_lhs, mx_rhs, np_lhs, np_rhs, dtype): argument 1952 np_rhs = mx_rhs.asnumpy() [all …]
|