Searched refs:a_dtype (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_matmul_op_test.py | 137 def _testGradients(self, tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype, delta, argument
146 tf_a = math_ops.cast(a, a_dtype) if a_dtype != dtypes.float32 else a
174 for a_dtype in (dtypes.float32, dtypes.bfloat16):
178 if a_dtype == dtypes.bfloat16 and b_dtype == dtypes.bfloat16:
183 self._testGradients(tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype,
|