()
| 55 | |
| 56 | |
| 57 | def make_dummy_features_labels(): |
| 58 | # XLA CPU/GPU backend doesn't support guaranteed constant, thus use dataset |
| 59 | # container to work around. |
| 60 | features_dataset = dataset_ops.Dataset.from_tensors( |
| 61 | constant_op.constant(_EXPECTED_FEATURE)).repeat(10) |
| 62 | features_op = features_dataset.make_one_shot_iterator().get_next() |
| 63 | labels_dataset = dataset_ops.Dataset.from_tensors( |
| 64 | constant_op.constant(_EXPECTED_LABEL)).repeat(10) |
| 65 | labels_op = labels_dataset.make_one_shot_iterator().get_next() |
| 66 | return features_op, labels_op |
| 67 | |
| 68 | |
| 69 | class XlaDecoratorTest(test.TestCase, parameterized.TestCase): |
no test coverage detected