Commit 7497fca0 authored by Yong Tang's avatar Yong Tang Committed by Martin Wicke
Browse files

Implement shape constraints for `Cross` (#14156)



* Implement shape constraints for `Cross`

This fix implements shape constrainsts for `Cross`, as
was specified in the TODO:
```
    // * Both inputs have the same shape.
    // * Input rank >= 1.
    // * input_shape[-1] == 3.
```

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add unit test for shape constraint and inference for `Cross`

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent a9bd6d67
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment