Home
last modified time | relevance | path

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 Dtest_sparse_operator.py1910 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 Dtest_sparse_operator.py1910 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 …]