Skip to content

Commit 435599f

Browse files
tensorflower-gardenertensorflower-gardener
committed
Adjust test sizes
PiperOrigin-RevId: 158565259
1 parent 0058c1f commit 435599f

44 files changed

Lines changed: 144 additions & 0 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎tensorflow/cc/BUILD‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ cc_library(
3434

3535
tf_cc_test(
3636
name = "framework_gradients_test",
37+
size = "small",
3738
srcs = ["framework/gradients_test.cc"],
3839
deps = [
3940
":cc_ops",
@@ -69,6 +70,7 @@ cc_library(
6970

7071
tf_cc_test(
7172
name = "framework_gradient_checker_test",
73+
size = "small",
7274
srcs = ["framework/gradient_checker_test.cc"],
7375
deps = [
7476
":cc_ops",
@@ -150,6 +152,7 @@ cc_library_with_android_deps(
150152

151153
tf_cc_test(
152154
name = "framework_scope_test",
155+
size = "small",
153156
srcs = ["framework/scope_test.cc"],
154157
deps = [
155158
":ops",
@@ -179,6 +182,7 @@ cc_library_with_android_deps(
179182

180183
tf_cc_test(
181184
name = "client_client_session_test",
185+
size = "small",
182186
srcs = ["client/client_session_test.cc"],
183187
deps = [
184188
":cc_ops",
@@ -213,6 +217,7 @@ cc_library_with_android_deps(
213217

214218
tf_cc_test(
215219
name = "ops_const_op_test",
220+
size = "small",
216221
srcs = ["ops/const_op_test.cc"],
217222
deps = [
218223
":const_op",
@@ -247,6 +252,7 @@ cc_library(
247252

248253
tf_cc_test(
249254
name = "gradients_array_grad_test",
255+
size = "small",
250256
srcs = ["gradients/array_grad_test.cc"],
251257
deps = [
252258
":array_grad",
@@ -277,6 +283,7 @@ cc_library(
277283

278284
tf_cc_test(
279285
name = "gradients_math_grad_test",
286+
size = "small",
280287
srcs = ["gradients/math_grad_test.cc"],
281288
deps = [
282289
":cc_ops",
@@ -307,6 +314,7 @@ cc_library(
307314

308315
tf_cc_test(
309316
name = "gradients_nn_grad_test",
317+
size = "small",
310318
srcs = ["gradients/nn_grad_test.cc"],
311319
deps = [
312320
":cc_ops",
@@ -355,6 +363,7 @@ tf_gen_op_wrappers_cc(
355363

356364
tf_cc_test(
357365
name = "framework_cc_ops_test",
366+
size = "small",
358367
srcs = ["framework/cc_ops_test.cc"],
359368
deps = [
360369
":cc_ops",
@@ -550,6 +559,7 @@ cc_library(
550559

551560
tf_cc_test(
552561
name = "coordinator_test",
562+
size = "small",
553563
srcs = ["training/coordinator_test.cc"],
554564
deps = [
555565
":cc_ops",

‎tensorflow/compiler/aot/BUILD‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ cc_library(
2020

2121
cc_test(
2222
name = "runtime_test",
23+
size = "small",
2324
srcs = ["runtime_test.cc"],
2425
deps = [
2526
":runtime",
@@ -88,6 +89,7 @@ cc_library(
8889

8990
cc_test(
9091
name = "codegen_test",
92+
size = "small",
9193
srcs = ["codegen_test.cc"],
9294
data = ["codegen_test_h.golden"],
9395
deps = [
@@ -101,6 +103,7 @@ cc_test(
101103

102104
cc_test(
103105
name = "tfcompile_util_test",
106+
size = "small",
104107
srcs = ["tfcompile_util_test.cc"],
105108
deps = [
106109
":tfcompile_lib",

‎tensorflow/compiler/tf2xla/BUILD‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ cc_library(
9292

9393
cc_test(
9494
name = "xla_compiler_test",
95+
size = "small",
9596
srcs = ["xla_compiler_test.cc"],
9697
deps = [
9798
":xla_compiler",
@@ -113,6 +114,7 @@ cc_test(
113114

114115
cc_test(
115116
name = "str_util_test",
117+
size = "small",
116118
srcs = [
117119
"str_util_test.cc",
118120
],
@@ -126,6 +128,7 @@ cc_test(
126128

127129
cc_test(
128130
name = "literal_util_test",
131+
size = "small",
129132
srcs = [
130133
"literal_util_test.cc",
131134
],

‎tensorflow/compiler/xla/BUILD‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -194,6 +194,7 @@ cc_library(
194194

195195
cc_test(
196196
name = "util_test",
197+
size = "small",
197198
srcs = ["util_test.cc"],
198199
deps = [
199200
":test",
@@ -234,6 +235,7 @@ cc_library(
234235

235236
cc_test(
236237
name = "shape_util_test",
238+
size = "small",
237239
srcs = ["shape_util_test.cc"],
238240
deps = [
239241
":shape_util",
@@ -248,6 +250,7 @@ cc_test(
248250

249251
cc_test(
250252
name = "layout_util_test",
253+
size = "small",
251254
srcs = ["layout_util_test.cc"],
252255
deps = [
253256
":shape_util",
@@ -260,6 +263,7 @@ cc_test(
260263

261264
cc_test(
262265
name = "index_util_test",
266+
size = "small",
263267
srcs = ["index_util_test.cc"],
264268
deps = [
265269
":shape_util",
@@ -289,6 +293,7 @@ cc_library(
289293

290294
cc_test(
291295
name = "literal_util_test",
296+
size = "small",
292297
srcs = ["literal_util_test.cc"],
293298
deps = [
294299
":array3d",
@@ -339,6 +344,7 @@ cc_library(
339344

340345
cc_test(
341346
name = "array2d_test",
347+
size = "small",
342348
srcs = ["array2d_test.cc"],
343349
deps = [
344350
":array2d",
@@ -359,6 +365,7 @@ cc_library(
359365

360366
cc_test(
361367
name = "array3d_test",
368+
size = "small",
362369
srcs = ["array3d_test.cc"],
363370
deps = [
364371
":array3d",
@@ -381,6 +388,7 @@ cc_library(
381388

382389
cc_test(
383390
name = "array4d_test",
391+
size = "small",
384392
srcs = ["array4d_test.cc"],
385393
deps = [
386394
":array4d",
@@ -448,6 +456,7 @@ cc_library(
448456

449457
cc_test(
450458
name = "text_literal_reader_test",
459+
size = "small",
451460
srcs = ["text_literal_reader_test.cc"],
452461
deps = [
453462
":literal_util",
@@ -478,6 +487,7 @@ cc_library(
478487

479488
cc_test(
480489
name = "text_literal_writer_test",
490+
size = "small",
481491
srcs = ["text_literal_writer_test.cc"],
482492
deps = [
483493
":literal_util",
@@ -505,6 +515,7 @@ cc_library(
505515

506516
cc_test(
507517
name = "shape_tree_test",
518+
size = "small",
508519
srcs = ["shape_tree_test.cc"],
509520
deps = [
510521
":shape_tree",
@@ -562,6 +573,7 @@ cc_library(
562573

563574
cc_test(
564575
name = "reference_util_test",
576+
size = "small",
565577
srcs = ["reference_util_test.cc"],
566578
deps = [
567579
":array2d",

‎tensorflow/compiler/xla/client/BUILD‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ cc_library(
4646

4747
cc_test(
4848
name = "padding_test",
49+
size = "small",
4950
srcs = ["padding_test.cc"],
5051
deps = [
5152
":padding",

‎tensorflow/compiler/xla/legacy_flags/BUILD‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ cc_library(
2929

3030
cc_test(
3131
name = "parse_flags_from_env_test",
32+
size = "small",
3233
srcs = ["parse_flags_from_env_test.cc"],
3334
deps =
3435
[

0 commit comments

Comments
 (0)