Mojo function
naive_epilogue
naive_epilogue[c_type: DType, c_shape: DimList[c_shape.values], *, elementwise_lambda_fn: elementwise_epilogue_type](c: NDBuffer[c_type, MutAnyOrigin, c_shape, DimList.create_unknown[2]()], ctx: DeviceContext)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!