Searched defs:AZp (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/python/mlir/dialects/linalg/opdsl/ops/ |
| H A D | core_named_ops.py | 413 quantized_matmul( A=TensorDef(T1, S.M, S.K), B=TensorDef(T2, S.K, S.N), AZp=ScalarDef(I32), BZp=ScalarDef(I32), C=TensorDef(U, S.M, S.N, output=True), ) global() argument
570 quantized_batch_matmul( A=TensorDef(T1, Batch, S.M, S.K), B=TensorDef(T2, Batch, S.K, S.N), AZp=ScalarDef(I32), BZp=ScalarDef(I32), C=TensorDef(U, Batch, S.M, S.N, output=True), ) global() argument
|