D | sparse_matmul_op_test.py | 47 sp_a=True, argument 59 a_is_sparse=sp_a, 111 for sp_a in [True, False]: 123 sp_a, 131 def _testGradients(self, tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype, delta, argument 149 a_is_sparse=sp_a, 165 for sp_a in [True, False]: 175 name = "sparse_matmul_%s_%s_%s_%s" % (tr_a, tr_b, sp_a, sp_b) 176 self._testGradients(tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype,
|