diff --git a/stdlib/public/TensorFlow/Gradients.swift b/stdlib/public/TensorFlow/Gradients.swift
index d6a9c2ef39654..794af47ba6021 100644
--- a/stdlib/public/TensorFlow/Gradients.swift
+++ b/stdlib/public/TensorFlow/Gradients.swift
@@ -455,7 +455,7 @@ func _vjpRsqrt<T : TensorFlowFloatingPoint>(
   _ x: Tensor<T>
 ) -> (Tensor<T>, (Tensor<T>) -> Tensor<T>) {
   let value = rsqrt(x)
-  return (value, { v in -v / 2 * value })
+  return (value, { v in -v / (2 * pow(x, 3 / 2))})
 }
 
 @inlinable