Searched defs:trans_a (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
H A Dcblas_reference.h44 const enum CBLAS_TRANSPOSE trans_a,
50 TFLITE_DCHECK(trans_a == CblasNoTrans);
43 cblas_sgemm(const enum CBLAS_ORDER order, const enum CBLAS_TRANSPOSE trans_a, const enum CBLAS_TRANSPOSE trans_b, const int m, const int n, const int k, const float alpha, const float *a, const int stride_a, const float *b, const int stride_b, const float beta, float *c, const int stride_c) argument
/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer_test.cc539 Output trans_a = ops::Transpose(s.WithOpName("trans_a"), a, perm); local
542 Output matmul = ops::MatMul(s.WithOpName("matmul"), trans_a, trans_b);
545 ops::SparseMatMul(s.WithOpName("matmul"), trans_a, trans_b);
548 ops::BatchMatMul(s.WithOpName("matmul"), trans_a, trans_b);
589 Output trans_a = ops::ConjugateTranspose(s.WithOpName("trans_a"), a, perm); local
591 Output matmul = ops::BatchMatMul(s.WithOpName("matmul"), trans_a, trans_b);

Completed in 167 milliseconds