)
py_library(
- name = "transformation_ops",
- srcs = [
- "batching.py",
- "enumerate_ops.py",
- "error_ops.py",
- "grouping.py",
- "interleave_ops.py",
- "resampling.py",
- "scan_ops.py",
- "sliding.py",
- "stats_ops.py",
- "threadpool.py",
- "unique.py",
+ name = "batching",
+ srcs = ["batching.py"],
+ srcs_version = "PY2AND3",
+ deps = [
++ "//tensorflow/contrib/framework:framework_py",
+ "//tensorflow/python:array_ops",
+ "//tensorflow/python:dataset_ops_gen",
+ "//tensorflow/python:dtypes",
+ "//tensorflow/python:framework_ops",
+ "//tensorflow/python:math_ops",
+ "//tensorflow/python:tensor_shape",
+ "//tensorflow/python:tensor_util",
+ "//tensorflow/python/data/ops:dataset_ops",
+ "//tensorflow/python/data/util:nest",
+ "//tensorflow/python/data/util:sparse",
],
+ )
+
+ py_library(
+ name = "enumerate_ops",
+ srcs = ["enumerate_ops.py"],
+ srcs_version = "PY2AND3",
+ deps = [
+ "//tensorflow/python:dtypes",
+ "//tensorflow/python/data/ops:dataset_ops",
+ ],
+ )
+
+ py_library(
+ name = "error_ops",
+ srcs = ["error_ops.py"],
srcs_version = "PY2AND3",
deps = [
":contrib_op_loader",
self.assertNotEqual(dtypes.int32, int)
self.assertNotEqual(dtypes.float64, 2.1)
+ def testPythonTypesConversion(self):
+ self.assertIs(dtypes.float32, dtypes.as_dtype(float))
+ self.assertIs(dtypes.bool, dtypes.as_dtype(bool))
+
+ def testReduce(self):
+ for enum in dtypes._TYPE_TO_STRING:
+ dtype = dtypes.DType(enum)
+ ctor, args = dtype.__reduce__()
+ self.assertEquals(ctor, dtypes.as_dtype)
+ self.assertEquals(args, (dtype.name,))
+ reconstructed = ctor(*args)
+ self.assertEquals(reconstructed, dtype)
+
+
if __name__ == "__main__":
googletest.main()