Searched defs:BatchMatmul (Results 1 – 2 of 2) sorted by relevance
488 static Graph* BatchMatmul(int b, int m, int k, int n, bool adjoint_a, in BatchMatmul() function
171 Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y) { in BatchMatmul() function