diff --git a/README.md b/README.md index 20c785f5..625aeda9 100644 --- a/README.md +++ b/README.md @@ -130,9 +130,9 @@ The autodiff stuff is all in mlops now, so you can focus on the raw operations ``` Buffer # class of memory on this device -unary_op (RELU, EXP, LOG, NEG, SIGN) # A -> A -binary_op (ADD, SUB, MUL, DIV, POW, A, CMQEQ) # A + B -> C -reduce_op (SUM, MAX) # A -> B (smaller size) +unary_op (RELU, EXP, LOG, NEG, SIGN) # A -> A +binary_op (ADD, SUB, MUL, DIV, POW, A, CMPEQ) # A + B -> C (broadcasting supported) +reduce_op (SUM, MAX) # A -> B (smaller size, B can have 1s in the shape) perm_axis # A -> A inner_slice # A -> B (different size) matmul # A + B -> C