1
0
forked from M-Labs/nac3

core/ndstrides: add TODO in np_reshape

This commit is contained in:
lyken 2024-07-30 15:40:41 +08:00
parent 3dc4b17310
commit fb54d5d112

View File

@ -1341,6 +1341,8 @@ impl<'a> BuiltinBuilder<'a> {
match prim {
PrimDef::FunNpReshape => {
// TODO: Support scalar inputs, e.g., `np.reshape(99, (1, 1, 1, 1))`
let new_ndim_ty = self.unifier.get_fresh_var(Some("NewNDim".into()), None);
let returned_ndarray_ty = make_ndarray_ty(
self.unifier,