Searched refs:transposeA (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
73 .setAttr("transpose_a", transposeA) in matmul()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_contraction_kernel.h | 146 const char transposeA = 'N';
155 mkldnn_status_t st = mkldnn_sgemm(&transposeA, &transposeB, &m, &n, &k,
|
| D | conv_grad_input_ops.cc | 239 const char transposeA = 'T'; // sgemm(A) == filter_data in operator ()() local
254 mkldnn_sgemm(&transposeA, &transposeB, &m, &n, &k, &alpha, filter_data, in operator ()()
|