Searched refs:BatchMatmul (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_test.cc | 488 static Graph* BatchMatmul(int b, int m, int k, int n, bool adjoint_a, in BatchMatmul() function
495 test::graph::BatchMatmul(g, test::graph::Constant(g, in0), in BatchMatmul()
540 test::Benchmark(#DEVICE, BatchMatmul<T>(B, M, K, N, TA, TB, TFTYPE)) \
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.h | 85 Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y);
|
| D | testlib.cc | 171 Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y) { in BatchMatmul() function
|