Searched refs:round_bfloat (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_matmul_op_test.py | 31 def RandMatrix(rows, cols, tr, round_bfloat=False): argument
34 rand_func = np.random.randint if round_bfloat else np.random.uniform
142 3, 2, tr_a, round_bfloat=True), dtype=dtypes.float32)
145 2, 4, tr_b, round_bfloat=True), dtype=dtypes.float32)
|