Home
last modified time | relevance | path

Searched refs:test_binary_operation (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/lib/core/
H A Dbfloat16_test.py63 def test_binary_operation(a, b, op, float_type): function
219 test_binary_operation(a, b, op=lambda a, b: a + b, float_type=float_type)
247 test_binary_operation(a, b, op=lambda a, b: a - b, float_type=float_type)
252 test_binary_operation(a, b, op=lambda a, b: a * b, float_type=float_type)
258 test_binary_operation(a, b, op=lambda a, b: a / b, float_type=float_type)
/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse.py5121 def test_binary_operation(self, layout, device, dtype, op): member in TestSparseAny