Open Bihaqo opened 5 years ago
Something like
@support_batches def matmul(a, b): ... matmul(a, b) matmul(batch_a, b) matmul(batch_a, batch_b)
Something like