Add vjps that don't use slicing or gather on shapes. (#556)
Add vjps that don't use slicing or gather on shapes. (#556)


tensorflow/swift-apis,tensorflow/swift-apis,tensorflow/swift-apis,tensorflow/swift-apis,tensorflow/swift-apis