chore: import upstream snapshot with attribution

tensorflow/python/kernel_tests/custom_ops/BUILD

@@ -0,0 +1,77 @@
+# Tests of TensorFlow custom ops written
+
+load("//tensorflow:tensorflow.bzl", "tf_custom_op_library")
+load("//tensorflow:tensorflow.default.bzl", "tf_py_strict_test")
+
+package(
+    # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
+    licenses = ["notice"],
+)
+
+# Custom op tests
+tf_custom_op_library(
+    name = "ackermann_op.so",
+    srcs = ["ackermann_op.cc"],
+)
+
+tf_py_strict_test(
+    name = "ackermann_test",
+    size = "small",
+    srcs = ["ackermann_test.py"],
+    data = [":ackermann_op.so"],
+    tags = [
+        "no_pip",
+        "notap",
+    ],
+    deps = [
+        "//tensorflow/python/framework:load_library",
+        "//tensorflow/python/framework:test_lib",
+        "//tensorflow/python/platform:client_testlib",
+        "//tensorflow/python/platform:resource_loader",
+    ],
+)
+
+tf_custom_op_library(
+    name = "duplicate_op.so",
+    srcs = ["duplicate_op.cc"],
+)
+
+tf_py_strict_test(
+    name = "duplicate_op_test",
+    size = "small",
+    srcs = ["duplicate_op_test.py"],
+    data = [":duplicate_op.so"],
+    tags = [
+        "no_pip",
+        "notap",
+    ],
+    deps = [
+        "//tensorflow/python/framework:load_library",
+        "//tensorflow/python/framework:test_lib",
+        "//tensorflow/python/ops:math_ops",
+        "//tensorflow/python/platform:client_testlib",
+        "//tensorflow/python/platform:resource_loader",
+    ],
+)
+
+tf_custom_op_library(
+    name = "invalid_op.so",
+    srcs = ["invalid_op.cc"],
+)
+
+tf_py_strict_test(
+    name = "invalid_op_test",
+    size = "small",
+    srcs = ["invalid_op_test.py"],
+    data = [":invalid_op.so"],
+    tags = [
+        "no_pip",
+        "notap",
+    ],
+    deps = [
+        "//tensorflow/python/framework:errors",
+        "//tensorflow/python/framework:load_library",
+        "//tensorflow/python/platform:client_testlib",
+        "//tensorflow/python/platform:resource_loader",
+    ],
+)

tensorflow/python/kernel_tests/custom_ops/ackermann_op.cc

@@ -0,0 +1,46 @@
+/* Copyright 2015 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.
+==============================================================================*/
+
+// An example Op.
+
+#include "tensorflow/core/framework/op.h"
+#include "tensorflow/core/framework/op_kernel.h"
+
+namespace tensorflow {
+
+REGISTER_OP("Ackermann")
+    .Output("ackermann: string")
+    .Doc(R"doc(
+Output a fact about the ackermann function.
+)doc");
+
+class AckermannOp : public OpKernel {
+ public:
+  explicit AckermannOp(OpKernelConstruction* context) : OpKernel(context) {}
+
+  void Compute(OpKernelContext* context) override {
+    // Output a scalar string.
+    Tensor* output_tensor = nullptr;
+    OP_REQUIRES_OK(context,
+                   context->allocate_output(0, TensorShape(), &output_tensor));
+    auto output = output_tensor->scalar<tstring>();
+
+    output() = "A(m, 0) == A(m-1, 1)";
+  }
+};
+
+REGISTER_KERNEL_BUILDER(Name("Ackermann").Device(DEVICE_CPU), AckermannOp);
+
+}  // namespace tensorflow

tensorflow/python/kernel_tests/custom_ops/ackermann_test.py

@@ -0,0 +1,37 @@
+# Copyright 2015 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.
+# ==============================================================================
+"""Tests for custom user ops."""
+import os
+
+from tensorflow.python.framework import load_library
+from tensorflow.python.framework import test_util
+from tensorflow.python.platform import resource_loader
+from tensorflow.python.platform import test
+
+
+class AckermannTest(test.TestCase):
+
+  @test_util.run_deprecated_v1
+  def testBasic(self):
+    library_filename = os.path.join(resource_loader.get_data_files_path(),
+                                    'ackermann_op.so')
+    ackermann = load_library.load_op_library(library_filename)
+
+    with self.cached_session():
+      self.assertEqual(ackermann.ackermann().eval(), b'A(m, 0) == A(m-1, 1)')
+
+
+if __name__ == '__main__':
+  test.main()

tensorflow/python/kernel_tests/custom_ops/duplicate_op.cc

@@ -0,0 +1,26 @@
+/* Copyright 2016 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.
+==============================================================================*/
+
+#include "tensorflow/core/framework/op.h"
+#include "tensorflow/core/framework/op_kernel.h"
+
+namespace tensorflow {
+
+REGISTER_OP("Add").Doc(R"doc(
+An op to test that duplicate registrations don't override previously
+registered ops.
+)doc");
+
+}  // namespace tensorflow

tensorflow/python/kernel_tests/custom_ops/duplicate_op_test.py

@@ -0,0 +1,38 @@
+# Copyright 2016 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.
+# ==============================================================================
+"""Tests for custom user ops."""
+import os
+
+from tensorflow.python.framework import load_library
+from tensorflow.python.framework import test_util
+from tensorflow.python.ops import math_ops
+from tensorflow.python.platform import resource_loader
+from tensorflow.python.platform import test
+
+
+class DuplicateOpTest(test.TestCase):
+
+  @test_util.run_deprecated_v1
+  def testBasic(self):
+    library_filename = os.path.join(resource_loader.get_data_files_path(),
+                                    'duplicate_op.so')
+    load_library.load_op_library(library_filename)
+
+    with self.cached_session():
+      self.assertEqual(math_ops.add(1, 41).eval(), 42)
+
+
+if __name__ == '__main__':
+  test.main()

tensorflow/python/kernel_tests/custom_ops/invalid_op.cc

@@ -0,0 +1,27 @@
+/* Copyright 2016 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.
+==============================================================================*/
+
+#include "tensorflow/core/framework/op.h"
+#include "tensorflow/core/framework/op_kernel.h"
+
+namespace tensorflow {
+
+REGISTER_OP("Invalid")
+    .Attr("invalid attr: int32")  // invalid since the name has a space.
+    .Doc(R"doc(
+An op to test that invalid ops do not successfully generate invalid python code.
+)doc");
+
+}  // namespace tensorflow

tensorflow/python/kernel_tests/custom_ops/invalid_op_test.py

@@ -0,0 +1,34 @@
+# Copyright 2016 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.
+# ==============================================================================
+"""Tests for custom user ops."""
+import os
+
+from tensorflow.python.framework import errors
+from tensorflow.python.framework import load_library
+from tensorflow.python.platform import resource_loader
+from tensorflow.python.platform import test
+
+
+class InvalidOpTest(test.TestCase):
+
+  def testBasic(self):
+    library_filename = os.path.join(resource_loader.get_data_files_path(),
+                                    'invalid_op.so')
+    with self.assertRaises(errors.InvalidArgumentError):
+      load_library.load_op_library(library_filename)
+
+
+if __name__ == '__main__':
+  test.main()