Searched refs:transposeA (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/ |
| D | TestUtil.java | 75 Graph g, String name, Output<T> a, Output<T> b, boolean transposeA, boolean transposeB) { in matmul() argument
79 .setAttr("transpose_a", transposeA) in matmul()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_contraction_kernel.h | 150 char transposeA = 'N', char transposeB = 'N') {
174 dnnl_sgemm(transposeB, transposeA, n, m, k, alpha, blockB, ldB, blockA,
211 char transposeA = 'N', char transposeB = 'N') {
246 dnnl_status_t st = dnnl_gemm_u8s8s32(transposeB, transposeA, offsetc, //
|
| D | conv_grad_input_ops.h | 327 const char transposeA = 'N'; // sgemm(A) == filter_data
342 dnnl_sgemm(transposeA, transposeB, m, n, k, alpha, out_data, ldA,
|