Commit e1066ba1 authored by Asim Shankar's avatar Asim Shankar Committed by TensorFlower Gardener
Browse files

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
parent 974bec95
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment