experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Adjust test sizes

Browse Source 
PiperOrigin-RevId: 158565259
This commit is contained in:
A. Unique TensorFlower 2017-06-09 14:45:22 -07:00 committed by TensorFlower Gardener
parent 0058c1f134
commit 435599f5d8
44 changed files with 144 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/cc/BUILD
View File

@ -34,6 +34,7 @@ cc_library(
tf_cc_test(
name = "framework_gradients_test",
size = "small",
srcs = ["framework/gradients_test.cc"],
deps = [
":cc_ops",
@ -69,6 +70,7 @@ cc_library(
tf_cc_test(
name = "framework_gradient_checker_test",
size = "small",
srcs = ["framework/gradient_checker_test.cc"],
deps = [
":cc_ops",
@ -150,6 +152,7 @@ cc_library_with_android_deps(
tf_cc_test(
name = "framework_scope_test",
size = "small",
srcs = ["framework/scope_test.cc"],
deps = [
":ops",
@ -179,6 +182,7 @@ cc_library_with_android_deps(
tf_cc_test(
name = "client_client_session_test",
size = "small",
srcs = ["client/client_session_test.cc"],
deps = [
":cc_ops",
@ -213,6 +217,7 @@ cc_library_with_android_deps(
tf_cc_test(
name = "ops_const_op_test",
size = "small",
srcs = ["ops/const_op_test.cc"],
deps = [
":const_op",
@ -247,6 +252,7 @@ cc_library(
tf_cc_test(
name = "gradients_array_grad_test",
size = "small",
srcs = ["gradients/array_grad_test.cc"],
deps = [
":array_grad",
@ -277,6 +283,7 @@ cc_library(
tf_cc_test(
name = "gradients_math_grad_test",
size = "small",
srcs = ["gradients/math_grad_test.cc"],
deps = [
":cc_ops",
@ -307,6 +314,7 @@ cc_library(
tf_cc_test(
name = "gradients_nn_grad_test",
size = "small",
srcs = ["gradients/nn_grad_test.cc"],
deps = [
":cc_ops",
@ -355,6 +363,7 @@ tf_gen_op_wrappers_cc(
tf_cc_test(
name = "framework_cc_ops_test",
size = "small",
srcs = ["framework/cc_ops_test.cc"],
deps = [
":cc_ops",
@ -550,6 +559,7 @@ cc_library(
tf_cc_test(
name = "coordinator_test",
size = "small",
srcs = ["training/coordinator_test.cc"],
deps = [
":cc_ops",

3
tensorflow/compiler/aot/BUILD
View File

@ -20,6 +20,7 @@ cc_library(
cc_test(
name = "runtime_test",
size = "small",
srcs = ["runtime_test.cc"],
deps = [
":runtime",
@ -88,6 +89,7 @@ cc_library(
cc_test(
name = "codegen_test",
size = "small",
srcs = ["codegen_test.cc"],
data = ["codegen_test_h.golden"],
deps = [
@ -101,6 +103,7 @@ cc_test(
cc_test(
name = "tfcompile_util_test",
size = "small",
srcs = ["tfcompile_util_test.cc"],
deps = [
":tfcompile_lib",

3
tensorflow/compiler/tf2xla/BUILD
View File

@ -92,6 +92,7 @@ cc_library(
cc_test(
name = "xla_compiler_test",
size = "small",
srcs = ["xla_compiler_test.cc"],
deps = [
":xla_compiler",
@ -113,6 +114,7 @@ cc_test(
cc_test(
name = "str_util_test",
size = "small",
srcs = [
"str_util_test.cc",
],
@ -126,6 +128,7 @@ cc_test(
cc_test(
name = "literal_util_test",
size = "small",
srcs = [
"literal_util_test.cc",
],

12
tensorflow/compiler/xla/BUILD
View File

@ -194,6 +194,7 @@ cc_library(
cc_test(
name = "util_test",
size = "small",
srcs = ["util_test.cc"],
deps = [
":test",
@ -234,6 +235,7 @@ cc_library(
cc_test(
name = "shape_util_test",
size = "small",
srcs = ["shape_util_test.cc"],
deps = [
":shape_util",
@ -248,6 +250,7 @@ cc_test(
cc_test(
name = "layout_util_test",
size = "small",
srcs = ["layout_util_test.cc"],
deps = [
":shape_util",
@ -260,6 +263,7 @@ cc_test(
cc_test(
name = "index_util_test",
size = "small",
srcs = ["index_util_test.cc"],
deps = [
":shape_util",
@ -289,6 +293,7 @@ cc_library(
cc_test(
name = "literal_util_test",
size = "small",
srcs = ["literal_util_test.cc"],
deps = [
":array3d",
@ -339,6 +344,7 @@ cc_library(
cc_test(
name = "array2d_test",
size = "small",
srcs = ["array2d_test.cc"],
deps = [
":array2d",
@ -359,6 +365,7 @@ cc_library(
cc_test(
name = "array3d_test",
size = "small",
srcs = ["array3d_test.cc"],
deps = [
":array3d",
@ -381,6 +388,7 @@ cc_library(
cc_test(
name = "array4d_test",
size = "small",
srcs = ["array4d_test.cc"],
deps = [
":array4d",
@ -448,6 +456,7 @@ cc_library(
cc_test(
name = "text_literal_reader_test",
size = "small",
srcs = ["text_literal_reader_test.cc"],
deps = [
":literal_util",
@ -478,6 +487,7 @@ cc_library(
cc_test(
name = "text_literal_writer_test",
size = "small",
srcs = ["text_literal_writer_test.cc"],
deps = [
":literal_util",
@ -505,6 +515,7 @@ cc_library(
cc_test(
name = "shape_tree_test",
size = "small",
srcs = ["shape_tree_test.cc"],
deps = [
":shape_tree",
@ -562,6 +573,7 @@ cc_library(
cc_test(
name = "reference_util_test",
size = "small",
srcs = ["reference_util_test.cc"],
deps = [
":array2d",

1
tensorflow/compiler/xla/client/BUILD
View File

@ -46,6 +46,7 @@ cc_library(
cc_test(
name = "padding_test",
size = "small",
srcs = ["padding_test.cc"],
deps = [
":padding",

1
tensorflow/compiler/xla/legacy_flags/BUILD
View File

@ -29,6 +29,7 @@ cc_library(
cc_test(
name = "parse_flags_from_env_test",
size = "small",
srcs = ["parse_flags_from_env_test.cc"],
deps =
[

33
tensorflow/compiler/xla/service/BUILD
View File

@ -56,6 +56,7 @@ cc_library(
cc_test(
name = "shape_inference_test",
size = "small",
srcs = ["shape_inference_test.cc"],
deps = [
":shape_inference",
@ -70,6 +71,7 @@ cc_test(
cc_test(
name = "hlo_opcode_test",
size = "small",
srcs = ["hlo_opcode_test.cc"],
deps = [
":hlo",
@ -100,6 +102,7 @@ cc_library(
cc_test(
name = "hlo_evaluator_test",
size = "small",
srcs = ["hlo_evaluator_test.cc"],
deps = [
":hlo",
@ -168,6 +171,7 @@ cc_library(
cc_test(
name = "hlo_matchers_test",
size = "small",
srcs = ["hlo_matchers_test.cc"],
deps = [
":hlo_matchers",
@ -188,6 +192,7 @@ cc_library(
cc_test(
name = "hlo_instruction_test",
size = "small",
srcs = ["hlo_instruction_test.cc"],
deps = [
":hlo",
@ -215,6 +220,7 @@ cc_library(
cc_test(
name = "call_graph_test",
size = "small",
srcs = ["call_graph_test.cc"],
deps = [
":call_graph",
@ -247,6 +253,7 @@ cc_library(
cc_test(
name = "flatten_call_graph_test",
size = "small",
srcs = ["flatten_call_graph_test.cc"],
deps = [
":call_graph",
@ -287,6 +294,7 @@ cc_library(
cc_test(
name = "user_computation_test",
size = "small",
srcs = ["user_computation_test.cc"],
deps = [
":hlo_matchers",
@ -641,6 +649,7 @@ cc_library(
cc_test(
name = "liveness_util_test",
size = "small",
srcs = ["liveness_util_test.cc"],
deps = [
":hlo",
@ -675,6 +684,7 @@ cc_library(
cc_test(
name = "buffer_liveness_test",
size = "small",
srcs = ["buffer_liveness_test.cc"],
deps = [
":buffer_liveness",
@ -717,6 +727,7 @@ cc_library(
cc_test(
name = "buffer_assignment_test",
size = "small",
srcs = ["buffer_assignment_test.cc"],
deps = [
":buffer_assignment",
@ -740,6 +751,7 @@ cc_test(
cc_test(
name = "heap_simulator_test",
size = "small",
srcs = ["heap_simulator_test.cc"],
deps = [
":hlo",
@ -784,6 +796,7 @@ cc_library(
cc_test(
name = "hlo_ordering_test",
size = "small",
srcs = ["hlo_ordering_test.cc"],
deps = [
":hlo",
@ -820,6 +833,7 @@ cc_library(
cc_test(
name = "instruction_fusion_test",
size = "small",
srcs = ["instruction_fusion_test.cc"],
deps = [
":hlo_matchers",
@ -850,6 +864,7 @@ cc_library(
cc_test(
name = "algebraic_simplifier_test",
size = "small",
srcs = ["algebraic_simplifier_test.cc"],
deps = [
":algebraic_simplifier",
@ -883,6 +898,7 @@ cc_library(
cc_test(
name = "reshape_mover_test",
size = "small",
srcs = ["reshape_mover_test.cc"],
deps = [
":hlo",
@ -916,6 +932,7 @@ cc_library(
cc_test(
name = "inliner_test",
size = "small",
srcs = ["inliner_test.cc"],
deps = [
":hlo",
@ -995,6 +1012,7 @@ cc_library(
cc_test(
name = "transfer_manager_test",
size = "small",
srcs = ["transfer_manager_test.cc"],
deps = [
":cpu_transfer_manager",
@ -1032,6 +1050,7 @@ cc_library(
cc_test(
name = "hlo_cost_analysis_test",
size = "small",
srcs = ["hlo_cost_analysis_test.cc"],
deps = [
":computation_tracker",
@ -1072,6 +1091,7 @@ cc_library(
cc_test(
name = "hlo_computation_test",
size = "small",
srcs = ["hlo_computation_test.cc"],
deps = [
":hlo",
@ -1102,6 +1122,7 @@ cc_binary(
cc_test(
name = "hlo_module_test",
size = "small",
srcs = ["hlo_module_test.cc"],
deps = [
":hlo",
@ -1159,6 +1180,7 @@ cc_library(
cc_test(
name = "hlo_dataflow_analysis_test",
size = "small",
srcs = ["hlo_dataflow_analysis_test.cc"],
deps = [
":hlo",
@ -1244,6 +1266,7 @@ cc_library(
cc_test(
name = "tuple_points_to_analysis_test",
size = "small",
srcs = ["tuple_points_to_analysis_test.cc"],
deps = [
":hlo",
@ -1327,6 +1350,7 @@ cc_library(
cc_test(
name = "copy_insertion_test",
size = "small",
srcs = ["copy_insertion_test.cc"],
deps = [
":copy_insertion",
@ -1391,6 +1415,7 @@ cc_library(
cc_test(
name = "hlo_rematerialization_test",
size = "small",
srcs = ["hlo_rematerialization_test.cc"],
deps = [
":hlo",
@ -1407,6 +1432,7 @@ cc_test(
cc_test(
name = "hlo_dce_test",
size = "small",
srcs = ["hlo_dce_test.cc"],
deps = [
":hlo",
@ -1426,6 +1452,7 @@ cc_test(
cc_test(
name = "layout_assignment_test",
size = "small",
srcs = ["layout_assignment_test.cc"],
deps = [
":algebraic_simplifier",
@ -1498,6 +1525,7 @@ cc_library(
cc_test(
name = "hlo_cse_test",
size = "small",
srcs = ["hlo_cse_test.cc"],
deps = [
":hlo",
@ -1534,6 +1562,7 @@ cc_library(
cc_test(
name = "hlo_constant_folding_test",
size = "small",
srcs = ["hlo_constant_folding_test.cc"],
deps = [
":hlo",
@ -1626,6 +1655,7 @@ cc_library(
cc_test(
name = "hlo_subcomputation_unification_test",
size = "small",
srcs = ["hlo_subcomputation_unification_test.cc"],
deps = [
":hlo",
@ -1654,6 +1684,7 @@ cc_library(
cc_test(
name = "hlo_tfgraph_builder_test",
size = "small",
srcs = ["hlo_tfgraph_builder_test.cc"],
deps = [
":hlo_tfgraph_builder",
@ -1700,6 +1731,7 @@ cc_library(
cc_test(
name = "transpose_folding_test",
size = "small",
srcs = ["transpose_folding_test.cc"],
deps = [
":hlo",
@ -1728,6 +1760,7 @@ cc_library(
cc_test(
name = "pool_test",
size = "small",
srcs = ["pool_test.cc"],
deps = [
":pool",

2
tensorflow/compiler/xla/service/cpu/BUILD
View File

@ -414,6 +414,7 @@ cc_test(
cc_test(
name = "infeed_manager_test",
size = "small",
srcs = ["infeed_manager_test.cc"],
deps = [
":cpu_runtime",
@ -506,6 +507,7 @@ cc_library(
cc_test(
name = "conv_canonicalization_test",
size = "small",
srcs = ["conv_canonicalization_test.cc"],
deps = [
":conv_canonicalization",

7
tensorflow/compiler/xla/service/gpu/BUILD
View File

@ -76,6 +76,7 @@ cc_library(
cc_test(
name = "stream_assignment_test",
size = "small",
srcs = [
"stream_assignment_test.cc",
],
@ -304,6 +305,7 @@ cc_library(
cc_test(
name = "convolution_folding_test",
size = "small",
srcs = ["convolution_folding_test.cc"],
deps = [
":convolution_folding",
@ -329,6 +331,7 @@ cc_library(
cc_test(
name = "instruction_fusion_test",
size = "small",
srcs = ["instruction_fusion_test.cc"],
deps = [
":instruction_fusion",
@ -367,6 +370,7 @@ cc_library(
cc_test(
name = "fusion_merger_test",
size = "small",
srcs = ["fusion_merger_test.cc"],
deps = [
":fusion_merger",
@ -474,6 +478,7 @@ cc_library(
cc_test(
name = "layout_assignment_test",
size = "small",
srcs = ["layout_assignment_test.cc"],
deps = [
":layout_assignment",
@ -502,6 +507,7 @@ cc_library(
cc_test(
name = "hlo_schedule_test",
size = "small",
srcs = [
"hlo_schedule_test.cc",
],
@ -532,6 +538,7 @@ cc_library(
cc_test(
name = "while_transformer_test",
size = "small",
srcs = ["while_transformer_test.cc"],
deps = [
":instruction_fusion",

3
tensorflow/contrib/batching/BUILD
View File

@ -27,6 +27,7 @@ cc_library(
cc_test(
name = "batch_scheduler_test",
size = "small",
srcs = [
"batch_scheduler_test.cc",
],
@ -60,6 +61,7 @@ cc_library(
cc_test(
name = "shared_batch_scheduler_test",
size = "small",
srcs = [
"shared_batch_scheduler_test.cc",
],
@ -83,6 +85,7 @@ cc_library(
cc_test(
name = "basic_batch_scheduler_test",
size = "small",
srcs = [
"basic_batch_scheduler_test.cc",
],

1
tensorflow/contrib/batching/util/BUILD
View File

@ -43,6 +43,7 @@ cc_library(
cc_test(
name = "periodic_function_test",
size = "small",
srcs = ["periodic_function_test.cc"],
deps = [
":periodic_function",

1
tensorflow/contrib/cluster_resolver/BUILD
View File

@ -35,6 +35,7 @@ py_library(
tf_py_test(
name = "cluster_resolver_py_test",
size = "small",
srcs = ["python/training/cluster_resolver_test.py"],
additional_deps = [
":cluster_resolver_py",

1
tensorflow/contrib/copy_graph/BUILD
View File

@ -27,6 +27,7 @@ py_library(
py_test(
name = "copy_test",
size = "small",
srcs = glob(["python/util/copy_test.py"]),
srcs_version = "PY2AND3",
deps = [

1
tensorflow/contrib/deprecated/BUILD
View File

@ -20,6 +20,7 @@ py_library(
py_test(
name = "summaries_test",
size = "small",
srcs = ["summaries_test.py"],
srcs_version = "PY2AND3",
deps = [

1
tensorflow/contrib/factorization/kernels/BUILD
View File

@ -51,6 +51,7 @@ cc_library(
cc_test(
name = "clustering_ops_test",
size = "small",
srcs = ["clustering_ops_test.cc"],
deps = [
":clustering_ops",

2
tensorflow/contrib/ffmpeg/BUILD
View File

@ -83,6 +83,7 @@ tf_gen_op_wrapper_py(
tf_py_test(
name = "decode_audio_op_test",
size = "small",
srcs = ["decode_audio_op_test.py"],
additional_deps = [
":ffmpeg_ops_py",
@ -97,6 +98,7 @@ tf_py_test(
tf_py_test(
name = "encode_audio_op_test",
size = "small",
srcs = ["encode_audio_op_test.py"],
additional_deps = [
":ffmpeg_ops_py",

2
tensorflow/contrib/framework/BUILD
View File

@ -234,6 +234,7 @@ py_test(
py_test(
name = "experimental_test",
size = "small",
srcs = ["python/framework/experimental_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -246,6 +247,7 @@ py_test(
py_test(
name = "tensor_util_test",
size = "small",
srcs = ["python/framework/tensor_util_test.py"],
srcs_version = "PY2AND3",
deps = [

7
tensorflow/contrib/graph_editor/BUILD
View File

@ -59,6 +59,7 @@ py_library(
py_test(
name = "util_test",
size = "small",
srcs = ["tests/util_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -73,6 +74,7 @@ py_test(
py_test(
name = "select_test",
size = "small",
srcs = ["tests/select_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -87,6 +89,7 @@ py_test(
py_test(
name = "match_test",
size = "small",
srcs = ["tests/match_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -102,6 +105,7 @@ py_test(
py_test(
name = "subgraph_test",
size = "small",
srcs = ["tests/subgraph_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -116,6 +120,7 @@ py_test(
py_test(
name = "reroute_test",
size = "small",
srcs = ["tests/reroute_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -131,6 +136,7 @@ py_test(
py_test(
name = "edit_test",
size = "small",
srcs = ["tests/edit_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -146,6 +152,7 @@ py_test(
py_test(
name = "transform_test",
size = "small",
srcs = ["tests/transform_test.py"],
srcs_version = "PY2AND3",
deps = [

1
tensorflow/contrib/layers/BUILD
View File

@ -182,6 +182,7 @@ py_test(
py_test(
name = "optimizers_test",
size = "small",
srcs = ["python/layers/optimizers_test.py"],
srcs_version = "PY2AND3",
deps = [

1
tensorflow/contrib/learn/BUILD
View File

@ -478,6 +478,7 @@ py_test(
py_test(
name = "tensor_signature_test",
size = "small",
srcs = ["python/learn/estimators/tensor_signature_test.py"],
srcs_version = "PY2AND3",
deps = [

1
tensorflow/contrib/metrics/BUILD
View File

@ -50,6 +50,7 @@ py_library(
py_test(
name = "classification_test",
size = "small",
srcs = ["python/metrics/classification_test.py"],
srcs_version = "PY2AND3",
deps = [

2
tensorflow/contrib/ndlstm/BUILD
View File

@ -54,6 +54,7 @@ tf_py_test(
tf_py_test(
name = "lstm2d_test",
size = "small",
srcs = ["python/lstm2d_test.py"],
additional_deps = [
":ndlstm",
@ -67,6 +68,7 @@ tf_py_test(
tf_py_test(
name = "misc_test",
size = "small",
srcs = ["python/misc_test.py"],
additional_deps = [
":ndlstm",

5
tensorflow/contrib/opt/BUILD
View File

@ -59,6 +59,7 @@ py_test(
py_test(
name = "external_optimizer_test",
size = "small",
srcs = ["python/training/external_optimizer_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -76,6 +77,7 @@ py_test(
py_test(
name = "moving_average_optimizer_test",
size = "small",
srcs = ["python/training/moving_average_optimizer_test.py"],
srcs_version = "PY2AND3",
tags = [
@ -112,6 +114,7 @@ py_test(
py_test(
name = "lazy_adam_optimizer_test",
size = "small",
srcs = ["python/training/lazy_adam_optimizer_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -129,6 +132,7 @@ py_test(
py_test(
name = "nadam_optimizer_test",
size = "small",
srcs = ["python/training/nadam_optimizer_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -146,6 +150,7 @@ py_test(
tf_py_test(
name = "drop_stale_gradient_optimizer_test",
size = "small",
srcs = ["python/training/drop_stale_gradient_optimizer_test.py"],
additional_deps = [
":opt_py",

1
tensorflow/contrib/saved_model/cc/saved_model/BUILD
View File

@ -37,6 +37,7 @@ cc_library(
cc_test(
name = "signature_def_utils_test",
size = "small",
srcs = ["signature_def_utils_test.cc"],
deps = [
":signature_def_utils",

1
tensorflow/contrib/session_bundle/BUILD
View File

@ -162,6 +162,7 @@ py_library(
py_test(
name = "gc_test",
size = "small",
srcs = [
"gc_test.py",
],

2
tensorflow/contrib/slim/BUILD
View File

@ -25,6 +25,7 @@ py_library(
py_test(
name = "evaluation_test",
size = "small",
srcs = ["python/slim/evaluation_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -175,6 +176,7 @@ py_library(
py_test(
name = "summaries_test",
size = "small",
srcs = ["python/slim/summaries_test.py"],
srcs_version = "PY2AND3",
deps = [

1
tensorflow/contrib/slim/python/slim/data/BUILD
View File

@ -59,6 +59,7 @@ py_library(
py_test(
name = "dataset_data_provider_test",
size = "small",
srcs = ["dataset_data_provider_test.py"],
srcs_version = "PY2AND3",
deps = [

1
tensorflow/contrib/specs/BUILD
View File

@ -53,6 +53,7 @@ tf_py_test(
tf_py_test(
name = "summaries_test",
size = "small",
srcs = ["python/summaries_test.py"],
additional_deps = [
":specs",

3
tensorflow/contrib/tfprof/python/tools/tfprof/BUILD
View File

@ -84,6 +84,7 @@ py_library(
tf_py_test(
name = "tfprof_logger_test",
size = "small",
srcs = ["tfprof_logger_test.py"],
additional_deps = [
":tfprof_logger",
@ -114,6 +115,7 @@ tf_py_wrap_cc(
py_test(
name = "print_model_analysis_test",
size = "small",
srcs = ["print_model_analysis_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -140,6 +142,7 @@ py_library(
py_test(
name = "pprof_profiler_test",
size = "small",
srcs = ["pprof_profiler_test.py"],
main = "pprof_profiler_test.py",
srcs_version = "PY2AND3",

1
tensorflow/contrib/xla_tf_graph/BUILD
View File

@ -41,6 +41,7 @@ cc_library(
cc_test(
name = "xla_tf_graph_util_test",
size = "small",
srcs = ["xla_tf_graph_util_test.cc"],
linkstatic = 1,
deps = [

3
tensorflow/core/grappler/BUILD
View File

@ -38,6 +38,7 @@ cc_library(
cc_test(
name = "utils_test",
size = "small",
srcs = ["utils_test.cc"],
deps = [
":utils",
@ -98,6 +99,7 @@ cc_library(
cc_test(
name = "grappler_item_test",
size = "small",
srcs = ["grappler_item_test.cc"],
deps = [
":grappler_item",
@ -110,6 +112,7 @@ cc_test(
cc_test(
name = "grappler_item_builder_test",
size = "small",
srcs = ["grappler_item_builder_test.cc"],
deps = [
":grappler_item_builder",

7
tensorflow/core/grappler/costs/BUILD
View File

@ -50,6 +50,7 @@ cc_library(
cc_test(
name = "graph_properties_test",
size = "small",
srcs = ["graph_properties_test.cc"],
args = ["--heap_check=local"], # The GPU tracer leaks memory
deps = [
@ -83,6 +84,7 @@ cc_library(
cc_test(
name = "graph_memory_test",
size = "small",
srcs = ["graph_memory_test.cc"],
args = ["--heap_check=local"], # The GPU tracer leaks memory
deps = [
@ -103,6 +105,7 @@ cc_library(
cc_test(
name = "robust_stats_test",
size = "small",
srcs = ["robust_stats_test.cc"],
deps = [
":robust_stats",
@ -153,6 +156,7 @@ cc_library(
cc_test(
name = "virtual_placer_test",
size = "small",
srcs = ["virtual_placer_test.cc"],
deps = [
":virtual_placer",
@ -185,6 +189,7 @@ cc_library(
cc_test(
name = "virtual_scheduler_test",
size = "small",
srcs = ["virtual_scheduler_test.cc"],
deps = [
":virtual_placer",
@ -232,6 +237,7 @@ cc_library(
cc_test(
name = "op_level_cost_estimator_test",
size = "small",
srcs = ["op_level_cost_estimator_test.cc"],
deps = [
":op_level_cost_estimator",
@ -264,6 +270,7 @@ cc_library(
cc_test(
name = "analytical_cost_estimator_test",
size = "small",
srcs = ["analytical_cost_estimator_test.cc"],
deps = [
":analytical_cost_estimator",

1
tensorflow/core/grappler/inputs/BUILD
View File

@ -30,6 +30,7 @@ cc_library(
cc_test(
name = "utils_test",
size = "small",
srcs = [
"utils_test.cc",
],

6
tensorflow/core/grappler/optimizers/BUILD
View File

@ -35,6 +35,7 @@ cc_library(
cc_test(
name = "static_schedule_test",
size = "small",
srcs = ["static_schedule_test.cc"],
deps = [
":static_schedule",
@ -71,6 +72,7 @@ cc_library(
cc_test(
name = "auto_parallel_test",
size = "small",
srcs = ["auto_parallel_test.cc"],
deps = [
":auto_parallel",
@ -106,6 +108,7 @@ cc_library(
cc_test(
name = "constant_folding_test",
size = "small",
srcs = ["constant_folding_test.cc"],
deps = [
":constant_folding",
@ -168,6 +171,7 @@ cc_library(
cc_test(
name = "model_pruner_test",
size = "small",
srcs = ["model_pruner_test.cc"],
deps = [
":model_pruner",
@ -201,6 +205,7 @@ cc_library(
cc_test(
name = "memory_optimizer_test",
size = "small",
srcs = ["memory_optimizer_test.cc"],
deps = [
":memory_optimizer",
@ -238,6 +243,7 @@ cc_library(
cc_test(
name = "layout_optimizer_test",
size = "small",
srcs = ["layout_optimizer_test.cc"],
deps = [
":layout_optimizer",

2
tensorflow/core/grappler/utils/BUILD
View File

@ -26,6 +26,7 @@ cc_library(
cc_test(
name = "scc_test",
size = "small",
srcs = ["scc_test.cc"],
deps = [
":scc",
@ -53,6 +54,7 @@ cc_library(
cc_test(
name = "topological_sort_test",
size = "small",
srcs = ["topological_sort_test.cc"],
deps = [
":topological_sort",

1
tensorflow/core/util/tensor_bundle/BUILD
View File

@ -64,6 +64,7 @@ cc_library(
cc_test(
name = "tensor_bundle_test",
size = "small",
srcs = ["tensor_bundle_test.cc"],
deps = [
":tensor_bundle",

1
tensorflow/examples/benchmark/BUILD
View File

@ -12,6 +12,7 @@ exports_files(["LICENSE"])
py_test(
name = "sample_benchmark",
size = "small",
srcs = ["sample_benchmark.py"],
srcs_version = "PY2AND3",
deps = [

1
tensorflow/python/estimator/BUILD
View File

@ -287,6 +287,7 @@ py_library(
py_test(
name = "parsing_utils_test",
size = "small",
srcs = ["canned/parsing_utils_test.py"],
srcs_version = "PY2AND3",
deps = [

2
tensorflow/python/tools/BUILD
View File

@ -189,6 +189,7 @@ py_binary(
py_test(
name = "print_selective_registration_header_test",
size = "small",
srcs = ["print_selective_registration_header_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -211,6 +212,7 @@ py_binary(
py_test(
name = "saved_model_cli_test",
size = "small",
srcs = ["saved_model_cli_test.py"],
data = [
"//tensorflow/cc/saved_model:saved_model_half_plus_two",

1
tensorflow/tools/api/tests/BUILD
View File

@ -13,6 +13,7 @@ exports_files([
py_test(
name = "api_compatibility_test",
size = "small",
srcs = ["api_compatibility_test.py"],
data = [
"//tensorflow/tools/api/golden:api_golden",

2
tensorflow/tools/common/BUILD
View File

@ -19,6 +19,7 @@ py_library(
py_test(
name = "public_api_test",
size = "small",
srcs = ["public_api_test.py"],
srcs_version = "PY2AND3",
deps = [
@ -35,6 +36,7 @@ py_library(
py_test(
name = "traverse_test",
size = "small",
srcs = ["traverse_test.py"],
srcs_version = "PY2AND3",
deps = [

1
tensorflow/tools/compatibility/BUILD
View File

@ -16,6 +16,7 @@ py_binary(
py_test(
name = "tf_upgrade_test",
size = "small",
srcs = ["tf_upgrade_test.py"],
srcs_version = "PY2AND3",
deps = [

1
tensorflow/tools/dist_test/scripts/BUILD
View File

@ -12,6 +12,7 @@ py_library(
py_test(
name = "k8s_tensorflow_test",
size = "small",
srcs = ["k8s_tensorflow_test.py"],
srcs_version = "PY2AND3",
deps = [

4
tensorflow/tools/tfprof/internal/BUILD
View File

@ -198,6 +198,7 @@ cc_library(
tf_cc_test(
name = "tfprof_show_test",
size = "small",
srcs = ["tfprof_show_test.cc"],
data = [
"testdata/ckpt.data-00000-of-00001",
@ -224,6 +225,7 @@ tf_cc_test(
tf_cc_test(
name = "tfprof_timeline_test",
size = "small",
srcs = ["tfprof_timeline_test.cc"],
data = [
"testdata/graph.pbtxt",
@ -295,6 +297,7 @@ cc_library(
tf_cc_test(
name = "tfprof_stats_test",
size = "small",
srcs = ["tfprof_stats_test.cc"],
data = [
"testdata/ckpt.data-00000-of-00001",
@ -333,6 +336,7 @@ cc_library(
tf_cc_test(
name = "tfprof_tensor_test",
size = "small",
srcs = ["tfprof_tensor_test.cc"],
data = [
"testdata/ckpt.data-00000-of-00001",