Searched refs:tr_a (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_matmul_op_test.py | 46 tr_a=False, argument
58 transpose_a=tr_a,
66 if tr_a:
115 for tr_a in [True, False]:
122 x = RandMatrix(n, k, tr_a)
127 tr_a,
137 def _testGradients(self, tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype, delta, argument
142 3, 2, tr_a, round_bfloat=True), dtype=dtypes.float32)
153 transpose_a=tr_a,
158 a, [2, 3] if tr_a else [3, 2],
[all …]
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | math_test.py | 231 for tr_a in (True, False):
235 shape_a = (5, 3) if tr_a else (3, 5)
249 return math_ops.matmul(a, b, transpose_a=tr_a, transpose_b=tr_b)
256 for tr_a in (True, False):
260 shape_a = (4, 5, 3) if tr_a else (4, 3, 5)
274 return math_ops.matmul(a, b, transpose_a=tr_a, transpose_b=tr_b)
|
| D | pfor.py | 1920 tr_a = pfor_input.get_attr("transpose_a")
1923 output = wrap(math_ops.matmul(a, b, adjoint_a=tr_a, adjoint_b=tr_b), True)
1926 if tr_a:
1963 prod = math_ops.matmul(a, b, transpose_a=tr_a)
|
| /external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/ |
| D | masked_matmul_benchmark.py | 116 tr_a=int(transpose_a),
|