reduce vjp for all and any (#2193)
This commit is contained in:
+1
-1
@@ -3548,7 +3548,7 @@ std::vector<array> Reduce::vjp(
|
||||
}
|
||||
|
||||
else {
|
||||
throw std::runtime_error("Reduce type VJP not yet implemented.");
|
||||
return {zeros_like(in, stream())};
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user