D | sparse_matmul_op_test.py | 49 sp_b=False, argument 61 b_is_sparse=sp_b) 118 for sp_b in [True, False]: 130 sp_b, 137 def _testGradients(self, tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype, delta, argument 156 b_is_sparse=sp_b) 173 for sp_b in [True, False]: 182 name = "sparse_matmul_%s_%s_%s_%s" % (tr_a, tr_b, sp_a, sp_b) 183 self._testGradients(tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype,
|