Update build file for keras/preprocessing.
PiperOrigin-RevId: 289360563 Change-Id: I494fd585d56adfb7928324a16cbb2ed67322820d
This commit is contained in:
		
							parent
							
								
									99a913907c
								
							
						
					
					
						commit
						f6273c2a78
					
				| @ -18,10 +18,6 @@ py_library( | |||||||
|         "estimator/__init__.py", |         "estimator/__init__.py", | ||||||
|         "keras_parameterized.py", |         "keras_parameterized.py", | ||||||
|         "ops.py", |         "ops.py", | ||||||
|         "preprocessing/__init__.py", |  | ||||||
|         "preprocessing/image.py", |  | ||||||
|         "preprocessing/sequence.py", |  | ||||||
|         "preprocessing/text.py", |  | ||||||
|         "testing_utils.py", |         "testing_utils.py", | ||||||
|     ], |     ], | ||||||
|     srcs_version = "PY2AND3", |     srcs_version = "PY2AND3", | ||||||
| @ -37,6 +33,7 @@ py_library( | |||||||
|         "//tensorflow/python/keras/mixed_precision/experimental:mixed_precision_experimental", |         "//tensorflow/python/keras/mixed_precision/experimental:mixed_precision_experimental", | ||||||
|         "//tensorflow/python/keras/optimizer_v2", |         "//tensorflow/python/keras/optimizer_v2", | ||||||
|         "//tensorflow/python/keras/premade", |         "//tensorflow/python/keras/premade", | ||||||
|  |         "//tensorflow/python/keras/preprocessing", | ||||||
|         "//tensorflow/python/keras/saving", |         "//tensorflow/python/keras/saving", | ||||||
|         "//tensorflow/python/keras/utils", |         "//tensorflow/python/keras/utils", | ||||||
|         "//tensorflow/python/keras/wrappers", |         "//tensorflow/python/keras/wrappers", | ||||||
| @ -1162,45 +1159,6 @@ cuda_py_test( | |||||||
|     ], |     ], | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| tf_py_test( |  | ||||||
|     name = "image_test", |  | ||||||
|     size = "medium", |  | ||||||
|     srcs = ["preprocessing/image_test.py"], |  | ||||||
|     python_version = "PY3", |  | ||||||
|     deps = [ |  | ||||||
|         ":keras", |  | ||||||
|         "//tensorflow/python:client_testlib", |  | ||||||
|         "//third_party/py/numpy", |  | ||||||
|         "@absl_py//absl/testing:parameterized", |  | ||||||
|     ], |  | ||||||
| ) |  | ||||||
| 
 |  | ||||||
| tf_py_test( |  | ||||||
|     name = "sequence_test", |  | ||||||
|     size = "small", |  | ||||||
|     srcs = ["preprocessing/sequence_test.py"], |  | ||||||
|     python_version = "PY3", |  | ||||||
|     deps = [ |  | ||||||
|         ":keras", |  | ||||||
|         "//tensorflow/python:client_testlib", |  | ||||||
|         "//third_party/py/numpy", |  | ||||||
|         "@absl_py//absl/testing:parameterized", |  | ||||||
|     ], |  | ||||||
| ) |  | ||||||
| 
 |  | ||||||
| tf_py_test( |  | ||||||
|     name = "text_test", |  | ||||||
|     size = "small", |  | ||||||
|     srcs = ["preprocessing/text_test.py"], |  | ||||||
|     python_version = "PY3", |  | ||||||
|     deps = [ |  | ||||||
|         ":keras", |  | ||||||
|         "//tensorflow/python:client_testlib", |  | ||||||
|         "//third_party/py/numpy", |  | ||||||
|         "@absl_py//absl/testing:parameterized", |  | ||||||
|     ], |  | ||||||
| ) |  | ||||||
| 
 |  | ||||||
| tf_py_test( | tf_py_test( | ||||||
|     name = "callbacks_test", |     name = "callbacks_test", | ||||||
|     size = "medium", |     size = "medium", | ||||||
|  | |||||||
							
								
								
									
										90
									
								
								tensorflow/python/keras/preprocessing/BUILD
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										90
									
								
								tensorflow/python/keras/preprocessing/BUILD
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,90 @@ | |||||||
|  | # Description: | ||||||
|  | #   Contains the Keras preprocessing layers (internal TensorFlow version). | ||||||
|  | 
 | ||||||
|  | load("//tensorflow:tensorflow.bzl", "tf_py_test") | ||||||
|  | 
 | ||||||
|  | package( | ||||||
|  |     default_visibility = ["//visibility:public"], | ||||||
|  |     licenses = ["notice"],  # Apache 2.0 | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | exports_files(["LICENSE"]) | ||||||
|  | 
 | ||||||
|  | py_library( | ||||||
|  |     name = "preprocessing", | ||||||
|  |     srcs = [ | ||||||
|  |         "__init__.py", | ||||||
|  |     ], | ||||||
|  |     deps = [ | ||||||
|  |         ":image", | ||||||
|  |         ":sequence", | ||||||
|  |         ":text", | ||||||
|  |     ], | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | py_library( | ||||||
|  |     name = "image", | ||||||
|  |     srcs = [ | ||||||
|  |         "image.py", | ||||||
|  |     ], | ||||||
|  |     deps = [ | ||||||
|  |         "//tensorflow/python:util", | ||||||
|  |         "//tensorflow/python/keras:backend", | ||||||
|  |         "//tensorflow/python/keras/utils:data_utils", | ||||||
|  |     ], | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | py_library( | ||||||
|  |     name = "sequence", | ||||||
|  |     srcs = [ | ||||||
|  |         "sequence.py", | ||||||
|  |     ], | ||||||
|  |     deps = [ | ||||||
|  |         "//tensorflow/python:util", | ||||||
|  |         "//tensorflow/python/keras/utils:data_utils", | ||||||
|  |     ], | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | py_library( | ||||||
|  |     name = "text", | ||||||
|  |     srcs = [ | ||||||
|  |         "text.py", | ||||||
|  |     ], | ||||||
|  |     deps = ["//tensorflow/python:util"], | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | tf_py_test( | ||||||
|  |     name = "image_test", | ||||||
|  |     size = "medium", | ||||||
|  |     srcs = ["image_test.py"], | ||||||
|  |     python_version = "PY3", | ||||||
|  |     deps = [ | ||||||
|  |         ":image", | ||||||
|  |         "//tensorflow/python:client_testlib", | ||||||
|  |         "//third_party/py/numpy", | ||||||
|  |     ], | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | tf_py_test( | ||||||
|  |     name = "sequence_test", | ||||||
|  |     size = "small", | ||||||
|  |     srcs = ["sequence_test.py"], | ||||||
|  |     python_version = "PY3", | ||||||
|  |     deps = [ | ||||||
|  |         ":sequence", | ||||||
|  |         "//tensorflow/python:client_testlib", | ||||||
|  |         "//third_party/py/numpy", | ||||||
|  |     ], | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | tf_py_test( | ||||||
|  |     name = "text_test", | ||||||
|  |     size = "small", | ||||||
|  |     srcs = ["text_test.py"], | ||||||
|  |     python_version = "PY3", | ||||||
|  |     deps = [ | ||||||
|  |         ":text", | ||||||
|  |         "//tensorflow/python:client_testlib", | ||||||
|  |         "//third_party/py/numpy", | ||||||
|  |     ], | ||||||
|  | ) | ||||||
| @ -36,16 +36,23 @@ py_library( | |||||||
|     ], |     ], | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
|  | py_library( | ||||||
|  |     name = "data_utils", | ||||||
|  |     srcs = ["data_utils.py"], | ||||||
|  |     srcs_version = "PY2AND3", | ||||||
|  |     deps = [":generic_utils"], | ||||||
|  | ) | ||||||
|  | 
 | ||||||
| py_library( | py_library( | ||||||
|     name = "engine_utils", |     name = "engine_utils", | ||||||
|     srcs = [ |     srcs = [ | ||||||
|         "conv_utils.py", |         "conv_utils.py", | ||||||
|         "data_utils.py", |  | ||||||
|         "io_utils.py", |         "io_utils.py", | ||||||
|         "losses_utils.py", |         "losses_utils.py", | ||||||
|     ], |     ], | ||||||
|     srcs_version = "PY2AND3", |     srcs_version = "PY2AND3", | ||||||
|     deps = [ |     deps = [ | ||||||
|  |         ":data_utils", | ||||||
|         "//tensorflow/python/keras:backend", |         "//tensorflow/python/keras:backend", | ||||||
|         "//tensorflow/python/ops/losses:loss_reduction", |         "//tensorflow/python/ops/losses:loss_reduction", | ||||||
|     ], |     ], | ||||||
|  | |||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user