• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Description:
2#   Contains the Keras engine API (internal TensorFlow version).
3
4# buildifier: disable=same-origin-load
5load("//tensorflow:tensorflow.bzl", "tf_py_test")
6
7# buildifier: disable=same-origin-load
8load("//tensorflow:tensorflow.bzl", "cuda_py_test")
9
10package(
11    # TODO(scottzhu): Remove non-keras deps from TF.
12    default_visibility = [
13        "//tensorflow/python:__pkg__",
14        "//tensorflow/python/feature_column:__pkg__",
15        "//tensorflow/python/keras:__subpackages__",
16    ],
17    licenses = ["notice"],
18)
19
20filegroup(
21    name = "all_py_srcs",
22    srcs = glob(["*.py"]),
23    visibility = ["//tensorflow/python/keras/google/private_tf_api_test:__pkg__"],
24)
25
26py_library(
27    name = "engine",
28    srcs = [
29        "__init__.py",
30        "compile_utils.py",
31        "functional.py",
32        "input_layer.py",
33        "partial_batch_padding_handler.py",
34        "saving.py",
35        "sequential.py",
36        "training.py",
37        "training_arrays_v1.py",
38        "training_distributed_v1.py",
39        "training_eager_v1.py",
40        "training_generator_v1.py",
41        "training_utils.py",
42        "training_utils_v1.py",
43        "training_v1.py",
44    ],
45    srcs_version = "PY3",
46    deps = [
47        ":base_layer",
48        ":base_preprocessing_layer",
49        ":data_adapter",
50        ":input_spec",
51        ":keras_tensor",
52        ":node",
53        "//tensorflow/python:py_checkpoint_reader",
54        "//tensorflow/python/data",
55        "//tensorflow/python/distribute:distribute_coordinator",
56        "//tensorflow/python/distribute:distribute_lib",
57        "//tensorflow/python/distribute:input_lib",
58        "//tensorflow/python/distribute:parameter_server_strategy",
59        "//tensorflow/python/distribute:parameter_server_strategy_v2",
60        "//tensorflow/python/distribute:reduce_util",
61        "//tensorflow/python/distribute/coordinator:cluster_coordinator",
62        "//tensorflow/python/eager:monitoring",
63        "//tensorflow/python/keras:activations",
64        "//tensorflow/python/keras:backend",
65        "//tensorflow/python/keras:callbacks",
66        "//tensorflow/python/keras:callbacks_v1",
67        "//tensorflow/python/keras:constraints",
68        "//tensorflow/python/keras:losses",
69        "//tensorflow/python/keras:metrics",
70        "//tensorflow/python/keras:optimizers",
71        "//tensorflow/python/keras:regularizers",
72        "//tensorflow/python/keras/distribute",
73        "//tensorflow/python/keras/distribute:distribute_coordinator_utils",
74        "//tensorflow/python/keras/initializers",
75        "//tensorflow/python/keras/mixed_precision:autocast_variable",
76        "//tensorflow/python/keras/mixed_precision:loss_scale_optimizer",
77        "//tensorflow/python/keras/mixed_precision:policy",
78        "//tensorflow/python/keras/saving",
79        "//tensorflow/python/keras/utils:engine_utils",
80        "//tensorflow/python/keras/utils:metrics_utils",
81        "//tensorflow/python/keras/utils:mode_keys",
82        "//tensorflow/python/keras/utils:tf_utils",
83        "//tensorflow/python/keras/utils:version_utils",
84        "//tensorflow/python/module",
85        "//tensorflow/python/ops/ragged:ragged_tensor",
86        "//tensorflow/python/ops/ragged:ragged_util",
87        "//tensorflow/python/profiler:trace",
88        "//tensorflow/python/saved_model:constants",
89        "//tensorflow/python/saved_model:loader",
90        "//tensorflow/python/tpu:tpu_lib",
91        "//tensorflow/python/training/tracking:data_structures",
92        "//tensorflow/tools/docs:doc_controls",
93    ],
94)
95
96py_library(
97    name = "base_layer_utils",
98    srcs = ["base_layer_utils.py"],
99    srcs_version = "PY3",
100    deps = [
101        "//tensorflow/python:array_ops",
102        "//tensorflow/python:auto_control_deps",
103        "//tensorflow/python:control_flow_v2_func_graphs",
104        "//tensorflow/python:dtypes",
105        "//tensorflow/python:framework_ops",
106        "//tensorflow/python:tf2",
107        "//tensorflow/python:util",
108        "//tensorflow/python:variables",
109        "//tensorflow/python/distribute:distribute_lib",
110        "//tensorflow/python/eager:context",
111        "//tensorflow/python/keras:backend",
112        "//tensorflow/python/keras/utils:tf_inspect",
113        "//tensorflow/python/keras/utils:tf_utils",
114    ],
115)
116
117py_library(
118    name = "base_layer",
119    srcs = [
120        "base_layer.py",
121        "base_layer_v1.py",
122    ],
123    srcs_version = "PY3",
124    deps = [
125        ":base_layer_utils",
126        ":input_spec",
127        ":node",
128        "//third_party/py/numpy",
129        "//tensorflow/python:array_ops",
130        "//tensorflow/python:auto_control_deps",
131        "//tensorflow/python:constant_op",
132        "//tensorflow/python:dtypes",
133        "//tensorflow/python:errors",
134        "//tensorflow/python:framework_ops",
135        "//tensorflow/python:func_graph",
136        "//tensorflow/python:math_ops",
137        "//tensorflow/python:platform",
138        "//tensorflow/python:resource_variable_ops",
139        "//tensorflow/python:sparse_tensor",
140        "//tensorflow/python:tensor_spec",
141        "//tensorflow/python:tensor_util",
142        "//tensorflow/python:tf2",
143        "//tensorflow/python/util:tf_export",
144        "//tensorflow/python:util",
145        "//tensorflow/python:variables",
146        "//tensorflow/python/autograph/core",
147        "//tensorflow/python/autograph/impl",
148        "//tensorflow/python/distribute:distribute_lib",
149        "//tensorflow/python/distribute:sharded_variable",
150        "//tensorflow/python/eager:context",
151        "//tensorflow/python/eager:execute",
152        "//tensorflow/python/eager:function",
153        "//tensorflow/python/eager:monitoring",
154        "//tensorflow/python/keras:backend",
155        "//tensorflow/python/keras:constraints",
156        "//tensorflow/python/keras/initializers",
157        # TODO(keras-team): Fix the cyclar deps between layer and metrics.
158        # "//tensorflow/python/keras:metrics",
159        "//tensorflow/python/keras:regularizers",
160        "//tensorflow/python/keras/mixed_precision:autocast_variable",
161        "//tensorflow/python/keras/mixed_precision:loss_scale_optimizer",
162        "//tensorflow/python/keras/mixed_precision:policy",
163        "//tensorflow/python/keras/saving",
164        "//tensorflow/python/keras/utils:generic_utils",
165        "//tensorflow/python/keras/utils:layer_utils",
166        "//tensorflow/python/keras/utils:object_identity",
167        "//tensorflow/python/keras/utils:tf_utils",
168        "//tensorflow/python/keras/utils:version_utils",
169        "//tensorflow/python/module",
170        "//tensorflow/python/ops/ragged:ragged_tensor",
171        "//tensorflow/python/training/tracking",
172        "//tensorflow/python/training/tracking:base",
173        "//tensorflow/python/training/tracking:data_structures",
174        "//tensorflow/python/training/tracking:layer_utils",
175        "//tensorflow/tools/docs:doc_controls",
176    ],
177)
178
179py_library(
180    name = "data_adapter",
181    srcs = ["data_adapter.py"],
182    srcs_version = "PY3",
183    deps = [
184        "//tensorflow/python:framework_ops",
185        "//tensorflow/python:util",
186        "//tensorflow/python/data/ops:dataset_ops",
187        "//tensorflow/python/keras/utils:dataset_creator",
188        "//tensorflow/python/keras/utils:engine_utils",
189        "//tensorflow/python/keras/utils:tf_utils",
190    ],
191)
192
193py_library(
194    name = "input_spec",
195    srcs = ["input_spec.py"],
196    srcs_version = "PY3",
197    deps = [
198        "//tensorflow/python:dtypes",
199        "//tensorflow/python:lib",
200        "//tensorflow/python:tensor_shape",
201        "//tensorflow/python:tensor_spec",
202        "//tensorflow/python/keras:backend",
203    ],
204)
205
206py_library(
207    name = "keras_tensor",
208    srcs = ["keras_tensor.py"],
209    srcs_version = "PY3",
210    deps = [
211        "//tensorflow/python:dtypes",
212        "//tensorflow/python:lib",
213        "//tensorflow/python:tensor_spec",
214        "//tensorflow/python/keras/utils:object_identity",
215    ],
216)
217
218py_library(
219    name = "base_preprocessing_layer",
220    srcs = [
221        "base_preprocessing_layer.py",
222    ],
223    srcs_version = "PY3",
224    deps = [
225        ":base_layer",
226        "//tensorflow/python/data",
227        "//tensorflow/python/eager:monitoring",
228        "//tensorflow/python/keras:backend",
229        "//tensorflow/python/module",
230    ],
231)
232
233py_library(
234    name = "node",
235    srcs = ["node.py"],
236    srcs_version = "PY3",
237    deps = [
238        ":base_layer_utils",
239        "//tensorflow/python:framework_ops",
240        "//tensorflow/python:platform",
241        "//tensorflow/python:tensor_util",
242        "//tensorflow/python:util",
243        "//tensorflow/python/keras:backend",
244        "//tensorflow/python/keras/utils:tf_utils",
245        "//third_party/py/numpy",
246    ],
247)
248
249tf_py_test(
250    name = "base_layer_utils_test",
251    srcs = ["base_layer_utils_test.py"],
252    python_version = "PY3",
253    tags = [
254        "nomac",  # TODO(mihaimaruseac): b/127695564
255    ],
256    deps = [
257        ":base_layer_utils",
258        "//tensorflow/python:client_testlib",
259        "//tensorflow/python:dtypes",
260        "//tensorflow/python:lookup_ops",
261        "//tensorflow/python/keras",
262        "//tensorflow/python/keras:backend",
263        "//tensorflow/python/keras:combinations",
264    ],
265)
266
267tf_py_test(
268    name = "data_adapter_test",
269    size = "medium",
270    srcs = ["data_adapter_test.py"],
271    python_version = "PY3",
272    shard_count = 4,
273    tags = [
274        "no_oss_py38",  # TODO(b/150615192)
275        "nomac",  # TODO(mihaimaruseac): b/127695564
276    ],
277    deps = [
278        ":data_adapter",
279        "//tensorflow/python:client_testlib",
280        "//tensorflow/python/keras",
281        "//third_party/py/numpy",
282    ],
283)
284
285tf_py_test(
286    name = "base_preprocessing_layer_test",
287    size = "medium",
288    srcs = ["base_preprocessing_layer_test.py"],
289    python_version = "PY3",
290    tags = [
291        "nomac",  # TODO(mihaimaruseac): b/127695564
292    ],
293    deps = [
294        ":base_preprocessing_layer",
295        "//tensorflow/python:client_testlib",
296        "//tensorflow/python/keras",
297        "@absl_py//absl/testing:parameterized",
298    ],
299)
300
301cuda_py_test(
302    name = "training_gpu_test",
303    size = "small",
304    srcs = ["training_gpu_test.py"],
305    python_version = "PY3",
306    tags = [
307        "nomac",  # TODO(mihaimaruseac): b/127695564
308    ],
309    xla_tags = [
310        "no_cuda_asan",  # times out
311    ],
312    deps = [
313        ":engine",
314        "//tensorflow/python:client_testlib",
315        "//tensorflow/python:framework_test_lib",
316        "//tensorflow/python/keras:backend",
317        "//tensorflow/python/keras:combinations",
318        "//tensorflow/python/keras/layers:convolutional",
319        "//third_party/py/numpy",
320        "@absl_py//absl/testing:parameterized",
321    ],
322)
323
324tf_py_test(
325    name = "correctness_test",
326    size = "medium",
327    srcs = ["correctness_test.py"],
328    python_version = "PY3",
329    shard_count = 2,
330    tags = [
331        "nomac",  # TODO(mihaimaruseac): b/127695564
332        "notsan",
333    ],
334    deps = [
335        "//tensorflow/python:client_testlib",
336        "//tensorflow/python/keras",
337        "//third_party/py/numpy",
338        "@absl_py//absl/testing:parameterized",
339    ],
340)
341
342tf_py_test(
343    name = "keras_tensor_test",
344    size = "small",
345    srcs = ["keras_tensor_test.py"],
346    python_version = "PY3",
347    tags = [
348        "nomac",  # TODO(mihaimaruseac): b/127695564
349    ],
350    deps = [
351        "//tensorflow/python:client_testlib",
352        "//tensorflow/python/keras",
353        "@absl_py//absl/testing:parameterized",
354    ],
355)
356
357tf_py_test(
358    name = "ragged_keras_tensor_test",
359    size = "small",
360    srcs = ["ragged_keras_tensor_test.py"],
361    python_version = "PY3",
362    tags = [
363        "nomac",  # TODO(mihaimaruseac): b/127695564
364    ],
365    deps = [
366        "//tensorflow/python:client_testlib",
367        "//tensorflow/python/keras",
368        "@absl_py//absl/testing:parameterized",
369    ],
370)
371
372tf_py_test(
373    name = "input_spec_test",
374    size = "small",
375    srcs = ["input_spec_test.py"],
376    python_version = "PY3",
377    tags = [
378        "nomac",  # TODO(mihaimaruseac): b/127695564
379    ],
380    deps = [
381        "//tensorflow/python:client_testlib",
382        "//tensorflow/python/keras",
383        "@absl_py//absl/testing:parameterized",
384    ],
385)
386
387tf_py_test(
388    name = "training_test",
389    size = "medium",
390    srcs = ["training_test.py"],
391    python_version = "PY3",
392    shard_count = 20,
393    tags = [
394        "manual",
395        "nomac",  # TODO(mihaimaruseac): b/127695564
396        "notsan",
397    ],
398    deps = [
399        ":engine",
400        "//tensorflow/python:array_ops",
401        "//tensorflow/python:client_testlib",
402        "//tensorflow/python:framework_ops",
403        "//tensorflow/python:framework_test_lib",
404        "//tensorflow/python:math_ops",
405        "//tensorflow/python:nn_ops",
406        "//tensorflow/python:resource_variable_ops",
407        "//tensorflow/python:sparse_ops",
408        "//tensorflow/python:state_ops",
409        "//tensorflow/python:tensor_shape",
410        "//tensorflow/python:training_lib",
411        "//tensorflow/python:variables",
412        "//tensorflow/python/data/ops:dataset_ops",
413        "//tensorflow/python/eager:context",
414        "//tensorflow/python/eager:function",
415        "//tensorflow/python/keras",
416        "//tensorflow/python/keras:backend",
417        "//tensorflow/python/keras:callbacks",
418        "//tensorflow/python/keras:combinations",
419        "//tensorflow/python/keras:losses",
420        "//tensorflow/python/keras:metrics",
421        "//tensorflow/python/keras:testing_utils",
422        "//tensorflow/python/keras/layers",
423        "//tensorflow/python/keras/utils:data_utils",
424        "//tensorflow/python/keras/utils:np_utils",
425        "//third_party/py/numpy",
426        "@absl_py//absl/testing:parameterized",
427    ],
428)
429
430tf_py_test(
431    name = "compile_utils_test",
432    size = "medium",
433    srcs = ["compile_utils_test.py"],
434    tags = [
435        "nomac",  # TODO(b/146226927)
436    ],
437    deps = [
438        "//tensorflow/python:client_testlib",
439        "//tensorflow/python/keras",
440        "//third_party/py/numpy",
441        "@absl_py//absl/testing:parameterized",
442    ],
443)
444
445tf_py_test(
446    name = "training_dataset_test",
447    size = "medium",
448    srcs = ["training_dataset_test.py"],
449    python_version = "PY3",
450    shard_count = 4,
451    tags = [
452        "nomac",  # TODO(mihaimaruseac): b/127695564
453    ],
454    deps = [
455        "//tensorflow/python:client_testlib",
456        "//tensorflow/python/keras",
457        "//third_party/py/numpy",
458        "@absl_py//absl/testing:parameterized",
459    ],
460)
461
462tf_py_test(
463    name = "training_arrays_test",
464    size = "medium",
465    srcs = ["training_arrays_test.py"],
466    python_version = "PY3",
467    tags = [
468        "nomac",  # TODO(mihaimaruseac): b/127695564
469    ],
470    deps = [
471        "//tensorflow/python:client_testlib",
472        "//tensorflow/python/data/ops:dataset_ops",
473        "//tensorflow/python/keras",
474        "//tensorflow/python/keras/layers",
475        "//third_party/py/numpy",
476        "@absl_py//absl/testing:parameterized",
477    ],
478)
479
480tf_py_test(
481    name = "training_generator_test",
482    size = "medium",
483    srcs = ["training_generator_test.py"],
484    python_version = "PY3",
485    shard_count = 6,
486    tags = [
487        "noasan",  # TODO(b/132183295): Re-enable this.
488        "nomac",  # TODO(b/140193633): Re-enable this.
489        "notsan",
490    ],
491    deps = [
492        ":engine",
493        "//tensorflow/python:client_testlib",
494        "//tensorflow/python:util",
495        "//tensorflow/python/data/ops:dataset_ops",
496        "//tensorflow/python/data/ops:iterator_ops",
497        "//tensorflow/python/eager:context",
498        "//tensorflow/python/keras",
499        "//tensorflow/python/keras:combinations",
500        "//tensorflow/python/keras:losses",
501        "//tensorflow/python/keras:metrics",
502        "//tensorflow/python/keras:testing_utils",
503        "//tensorflow/python/keras/layers",
504        "//tensorflow/python/keras/optimizer_v2",
505        "//tensorflow/python/keras/utils:data_utils",
506        "//third_party/py/numpy",
507        "@absl_py//absl/testing:parameterized",
508    ],
509)
510
511tf_py_test(
512    name = "training_integration_test",
513    size = "medium",
514    srcs = ["training_integration_test.py"],
515    python_version = "PY3",
516    shard_count = 30,
517    tags = [
518        "nomac",  # TODO(mihaimaruseac): b/127695564
519    ],
520    deps = [
521        "//tensorflow/python:client_testlib",
522        "//tensorflow/python/keras",
523        "//third_party/py/numpy",
524        "@absl_py//absl/testing:parameterized",
525    ],
526)
527
528tf_py_test(
529    name = "feature_columns_integration_test",
530    size = "medium",
531    srcs = ["feature_columns_integration_test.py"],
532    python_version = "PY3",
533    tags = [
534        "nomac",  # TODO(mihaimaruseac): b/127695564
535        "notsan",
536    ],
537    deps = [
538        "//tensorflow/python:client_testlib",
539        "//tensorflow/python/feature_column:feature_column_py",
540        "//tensorflow/python/keras",
541        "//third_party/py/numpy",
542        "@absl_py//absl/testing:parameterized",
543    ],
544)
545
546tf_py_test(
547    name = "training_eager_test",
548    size = "medium",
549    srcs = ["training_eager_test.py"],
550    python_version = "PY3",
551    tags = [
552        "nomac",  # TODO(mihaimaruseac): b/127695564
553        "notsan",
554    ],
555    deps = [
556        "//tensorflow/python:client_testlib",
557        "//tensorflow/python/keras",
558        "//third_party/py/numpy",
559        "@absl_py//absl/testing:parameterized",
560    ],
561)
562
563tf_py_test(
564    name = "training_utils_v1_test",
565    size = "medium",
566    srcs = ["training_utils_v1_test.py"],
567    python_version = "PY3",
568    tags = [
569        "no_oss",  # TODO(b/135021748) reenable
570        "notsan",
571    ],
572    deps = [
573        "//tensorflow/python:client_testlib",
574        "//tensorflow/python/keras",
575        "//third_party/py/numpy",
576        "@absl_py//absl/testing:parameterized",
577    ],
578)
579
580tf_py_test(
581    name = "input_layer_test",
582    size = "medium",
583    srcs = ["input_layer_test.py"],
584    python_version = "PY3",
585    shard_count = 3,
586    tags = [
587        "nomac",  # TODO(mihaimaruseac): b/127695564
588    ],
589    deps = [
590        ":base_layer",
591        ":engine",
592        "//tensorflow/python/keras",
593        "//tensorflow/python/keras:combinations",
594        "//tensorflow/python/keras:testing_utils",
595        "//tensorflow/python/keras/utils:layer_utils",
596    ],
597)
598
599tf_py_test(
600    name = "functional_test",
601    size = "medium",
602    srcs = ["functional_test.py"],
603    python_version = "PY3",
604    shard_count = 8,
605    tags = [
606        "no-internal-py3",
607        "no_rocm",
608        "nomac",  # TODO(mihaimaruseac): b/127695564
609    ],
610    deps = [
611        ":base_layer",
612        ":engine",
613        "//tensorflow/python:array_ops",
614        "//tensorflow/python:client_testlib",
615        "//tensorflow/python:constant_op",
616        "//tensorflow/python:dtypes",
617        "//tensorflow/python:framework_ops",
618        "//tensorflow/python:math_ops",
619        "//tensorflow/python:state_ops",
620        "//tensorflow/python:tensor_shape",
621        "//tensorflow/python/eager:context",
622        "//tensorflow/python/keras",
623        "//tensorflow/python/keras:backend",
624        "//tensorflow/python/keras:combinations",
625        "//tensorflow/python/keras:models",
626        "//tensorflow/python/keras:testing_utils",
627        "//tensorflow/python/keras/initializers",
628        "//tensorflow/python/keras/layers",
629        "//tensorflow/python/keras/utils:layer_utils",
630        "//tensorflow/python/keras/utils:tf_utils",
631        "//tensorflow/python/ops/ragged:ragged_factory_ops",
632        "//tensorflow/python/training/tracking:util",
633        "//third_party/py/numpy",
634    ],
635)
636
637tf_py_test(
638    name = "node_test",
639    size = "medium",
640    srcs = ["node_test.py"],
641    python_version = "PY3",
642    shard_count = 3,
643    tags = [
644        "nomac",  # TODO(mihaimaruseac): b/127695564
645    ],
646    deps = [
647        ":base_layer",
648        ":engine",
649        "//tensorflow/python/keras",
650        "//tensorflow/python/keras:testing_utils",
651        "//tensorflow/python/keras/utils:layer_utils",
652    ],
653)
654
655tf_py_test(
656    name = "base_layer_test",
657    size = "medium",
658    srcs = ["base_layer_test.py"],
659    python_version = "PY3",
660    shard_count = 8,
661    tags = [
662        "nomac",  # TODO(mihaimaruseac): b/127695564
663    ],
664    deps = [
665        ":base_layer",
666        ":engine",
667        "//tensorflow/python:array_ops",
668        "//tensorflow/python:client_testlib",
669        "//tensorflow/python:composite_tensor",
670        "//tensorflow/python:constant_op",
671        "//tensorflow/python:control_flow_ops",
672        "//tensorflow/python:dtypes",
673        "//tensorflow/python:framework_ops",
674        "//tensorflow/python:math_ops",
675        "//tensorflow/python:platform",
676        "//tensorflow/python:sparse_tensor",
677        "//tensorflow/python:state_ops",
678        "//tensorflow/python:summary_ops_v2",
679        "//tensorflow/python:tensor_array_ops",
680        "//tensorflow/python:tensor_spec",
681        "//tensorflow/python:type_spec",
682        "//tensorflow/python:util",
683        "//tensorflow/python:variables",
684        "//tensorflow/python/eager:context",
685        "//tensorflow/python/eager:def_function",
686        "//tensorflow/python/keras",
687        "//tensorflow/python/keras:backend",
688        "//tensorflow/python/keras:combinations",
689        "//tensorflow/python/keras:regularizers",
690        "//tensorflow/python/keras:testing_utils",
691        "//tensorflow/python/keras/layers",
692        "//tensorflow/python/keras/legacy_tf_layers:core",
693        "//tensorflow/python/keras/mixed_precision:policy",
694        "//tensorflow/python/keras/optimizer_v2",
695        "//tensorflow/python/keras/utils:tf_utils",
696        "//tensorflow/python/ops/ragged:ragged_tensor",
697        "//tensorflow/python/summary",
698        "//third_party/py/numpy",
699    ],
700)
701
702tf_py_test(
703    name = "control_flow_test",
704    size = "medium",
705    srcs = ["control_flow_test.py"],
706    python_version = "PY3",
707    shard_count = 8,
708    tags = [
709        "nomac",  # TODO(mihaimaruseac): b/127695564
710    ],
711    deps = [
712        "//tensorflow/python:client_testlib",
713        "//tensorflow/python/keras",
714        "//third_party/py/numpy",
715        "@absl_py//absl/testing:parameterized",
716    ],
717)
718
719tf_py_test(
720    name = "sequential_test",
721    size = "medium",
722    srcs = ["sequential_test.py"],
723    python_version = "PY3",
724    tags = [
725        "nomac",  # TODO(mihaimaruseac): b/127695564
726    ],
727    deps = [
728        "//tensorflow/python:client_testlib",
729        "//tensorflow/python/keras",
730        "//third_party/py/numpy",
731        "@absl_py//absl/testing:parameterized",
732    ],
733)
734
735tf_py_test(
736    name = "deferred_sequential_test",
737    size = "medium",
738    srcs = ["deferred_sequential_test.py"],
739    python_version = "PY3",
740    tags = [
741        "nomac",  # TODO(mihaimaruseac): b/127695564
742    ],
743    deps = [
744        "//tensorflow/python:client_testlib",
745        "//tensorflow/python/compat:v2_compat",
746        "//tensorflow/python/keras",
747        "//third_party/py/numpy",
748        "@absl_py//absl/testing:parameterized",
749    ],
750)
751