Searched refs:transposeB (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/ |
| D | TestUtil.java | 69 Graph g, String name, Output<T> a, Output<T> b, boolean transposeA, boolean transposeB) { in matmul() argument
74 .setAttr("transpose_b", transposeB) in matmul()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_contraction_kernel.h | 147 const char transposeB = 'N';
155 mkldnn_status_t st = mkldnn_sgemm(&transposeA, &transposeB, &m, &n, &k,
|
| D | conv_grad_input_ops.cc | 240 const char transposeB = 'N'; // sgemm(B) == out_data in operator ()() local
254 mkldnn_sgemm(&transposeA, &transposeB, &m, &n, &k, &alpha, filter_data, in operator ()()
|