1# Copyright 2017 The TensorFlow Authors. All Rights Reserved. 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14# ============================================================================== 15# Description: 16# Example TensorFlow models for CIFAR-10 17 18package( 19 default_visibility = [ 20 "//tensorflow:internal", 21 ], 22) 23 24licenses(["notice"]) # Apache 2.0 25 26py_library( 27 name = "cifar10_input", 28 srcs = ["cifar10_input.py"], 29 srcs_version = "PY2AND3", 30 deps = [ 31 "//tensorflow:tensorflow_py", 32 ], 33) 34 35py_library( 36 name = "cifar10_pruning", 37 srcs = ["cifar10_pruning.py"], 38 srcs_version = "PY2AND3", 39 deps = [ 40 ":cifar10_input", 41 "//tensorflow:tensorflow_py", 42 "//tensorflow/contrib/model_pruning:pruning", 43 ], 44) 45 46py_binary( 47 name = "cifar10_eval", 48 srcs = [ 49 "cifar10_eval.py", 50 ], 51 srcs_version = "PY2AND3", 52 deps = [ 53 ":cifar10_pruning", 54 "//tensorflow:tensorflow_py", 55 "//third_party/py/numpy", 56 ], 57) 58 59py_binary( 60 name = "cifar10_train", 61 srcs = [ 62 "cifar10_train.py", 63 ], 64 srcs_version = "PY2AND3", 65 deps = [ 66 ":cifar10_pruning", 67 "//tensorflow:tensorflow_py", 68 "//tensorflow/contrib/model_pruning:pruning", 69 ], 70) 71