aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/tf2xla/lib/random.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/tf2xla/lib/random.h')
-rw-r--r--tensorflow/compiler/tf2xla/lib/random.h2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/tf2xla/lib/random.h b/tensorflow/compiler/tf2xla/lib/random.h
index 39cbcf9c5e..2c573fd85b 100644
--- a/tensorflow/compiler/tf2xla/lib/random.h
+++ b/tensorflow/compiler/tf2xla/lib/random.h
@@ -28,7 +28,7 @@ namespace tensorflow {
//
// The "uniform" parameter must be an array of random numbers distributed in
// (0,1).
-xla::XlaOp TruncatedNormal(DataType dtype, xla::XlaOp uniform);
+xla::XlaOp TruncatedNormal(xla::XlaOp uniform);
} // namespace tensorflow