// Copyright 2018 The TensorFlow Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#if !COMPILING_TENSORFLOW_MODULE
import TensorFlow
#endif

// Rounds the values of a tensor to the nearest integer, element-wise.
func round<Scalar: BinaryFloatingPoint>(_ x: Tensor<Scalar>) -> Tensor<Scalar> {
    return Raw.round(x)
}

// Return a tensor with the same shape and contents as input.
func identity<Scalar>(_ x: Tensor<Scalar>) -> Tensor<Scalar> {
    return x
}

// Copyright 2018 The TensorFlow Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#if !COMPILING_TENSORFLOW_MODULE
import TensorFlow
#endif

// `round` Tensor Float or Double values to the to closest integral values with greater magnitude
// This is a manual definition.
func round<T : BinaryFloatingPoint>(_ t: Tensor<T>) -> Tensor<T> {
    return Raw.round(t)
}

--- 
+++ 
@@ -16,8 +16,12 @@
 import TensorFlow
 #endif
 
-// `round` Tensor Float or Double values to the to closest integral values with greater magnitude
-// This is a manual definition.
-func round<T : BinaryFloatingPoint>(_ t: Tensor<T>) -> Tensor<T> {
-    return Raw.round(t)
+// Rounds the values of a tensor to the nearest integer, element-wise.
+func round<Scalar: BinaryFloatingPoint>(_ x: Tensor<Scalar>) -> Tensor<Scalar> {
+    return Raw.round(x)
 }
+
+// Return a tensor with the same shape and contents as input.
+func identity<Scalar>(_ x: Tensor<Scalar>) -> Tensor<Scalar> {
+    return x
+}
Add identity function and update doc comments.
