ResourceScatterUpdate: Gracefully handle inconsistent indices and updates in
the kernel. With graph execution, consistency between the shapes of the arguments to ResourceScatterUpdate is validated by the shape inference functions at graph construction time. With eager execution, the shape inference logic isn't executed, so inconsistent arguments could be provided to the kernel, which would result in a segmentation fault prior to this change. As demonstrated by the added tests. PiperOrigin-RevId: 188729154
Loading
Please sign in to comment