Searched defs:trans_a (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
H A D | cblas_reference.h | 44 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 D | arithmetic_optimizer_test.cc | 539 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