Searched refs:trans_a (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 723 Output trans_a = ops::Transpose(s.WithOpName("trans_a"), a, perm); in TEST_F() local
729 matmul = ops::MatMul(matmul_op, trans_a, trans_b); in TEST_F()
731 matmul = ops::SparseMatMul(matmul_op, trans_a, trans_b); in TEST_F()
733 matmul = ops::BatchMatMul(matmul_op, trans_a, trans_b); in TEST_F()
735 matmul = ops::BatchMatMulV2(matmul_op, trans_a, trans_b); in TEST_F()
797 Output trans_a = ops::ConjugateTranspose(s.WithOpName("trans_a"), a, perm); in TEST_F() local
799 Output matmul = ops::BatchMatMul(s.WithOpName("matmul"), trans_a, trans_b); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 1552 bool trans_a; in FusedMatMulRewrite() local
1556 TF_CHECK_OK(GetNodeAttr(n->def(), "transpose_a", &trans_a)); in FusedMatMulRewrite()
1558 return !trans_a; in FusedMatMulRewrite()
|