Searched refs:tr_b (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_matmul_op_test.py | 47 tr_b=False, argument
59 transpose_b=tr_b,
68 if tr_b:
116 for tr_b in [True, False]:
123 y = RandMatrix(k, m, tr_b)
128 tr_b,
137 def _testGradients(self, tr_a, tr_b, sp_a, sp_b, a_dtype, b_dtype, delta, argument
145 2, 4, tr_b, round_bfloat=True), dtype=dtypes.float32)
154 transpose_b=tr_b,
162 b, [4, 2] if tr_b else [2, 4],
[all …]
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | math_test.py | 232 for tr_b in (True, False):
238 shape_b = (7, 5) if tr_b else (5, 7)
249 return math_ops.matmul(a, b, transpose_a=tr_a, transpose_b=tr_b)
257 for tr_b in (True, False):
263 shape_b = (4, 7, 5) if tr_b else (4, 5, 7)
274 return math_ops.matmul(a, b, transpose_a=tr_a, transpose_b=tr_b)
|
| D | pfor.py | 1921 tr_b = pfor_input.get_attr("transpose_b")
1923 output = wrap(math_ops.matmul(a, b, adjoint_a=tr_a, adjoint_b=tr_b), True)
1936 prod = math_ops.matmul(a, b, transpose_b=tr_b)
1940 if tr_b:
|
| /external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/ |
| D | masked_matmul_benchmark.py | 117 tr_b=int(transpose_b),
|