| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 214495925
|
|
|
|
|
|
|
|
| |
self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about:
* the fact that the session may be reused.
* the session is not closed even when doing a "with self.test_session()" statement.
PiperOrigin-RevId: 213326581
|
|
|
|
|
|
|
|
|
|
| |
- replacing `OpKernelContext` with newly introduced `DatasetContext` in `DatasetBase` constructor to make it possible to instantiate `DatasetBase` in places where an instance of `OpKernelContext` is not available
- replacing `dataset::MakeIteratorContext(OpKernelContext* ctx)` factory with `IteratorContext(OpKernelContext *ctx)` constructor.
- folding `GraphDatasetBase` into `DataseBase` and removing the default implementation of `AsGraphDefInternal`, making it the responsibility of the derived class to implement it to encourage/hint developers to provide serialization logic
PiperOrigin-RevId: 208560010
|
|
|
|
|
|
|
|
| |
This CL:
- changes the `OptimizeDataset` checkpointing logic to checkpoint the optimized dataset (as opposed to the original dataset + the optimizations, re-running optimization every time a checkpoint is restored)
- replaces `OpKernelContext` with newly introduced `SerializationContext` in the signature of `AsGraphDefInternal` to reduce the scope of the context and also simplify the logic for overriding the `FunctionLibraryDefinition` when optimizations take place
PiperOrigin-RevId: 208282562
|
| |
|
|
|
|
|
|
| |
TowerLocalVariables.
PiperOrigin-RevId: 203520287
|
|\ |
|
| |
| |
| |
| |
| |
| | |
By marking DebugString() as const we can make some error messages more descriptive. Because DatasetIterator marks the return value of the dataset() function const, DebugString() cannot be called.
PiperOrigin-RevId: 198796894
|
| |
| |
| |
| |
| |
| | |
status.
PiperOrigin-RevId: 198772254
|
|\| |
|
| |
| |
| |
| | |
PiperOrigin-RevId: 190878279
|
| |
| |
| |
| | |
PiperOrigin-RevId: 189945839
|
| |
| |
| |
| | |
PiperOrigin-RevId: 189258641
|
| |
| |
| |
| | |
PiperOrigin-RevId: 189231636
|
| | |
|
|/
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
* Fix build issue with KafkaDataset
This fix tries to address the issue raised in 17210 where
error of `NotFoundError: Op type not registered 'KafkaDataset' in binary.`
returned from kafka ops. The issue was that the inclusion of kafka ops
was removed due to the conflict merge from the other PR. This fix
fixes the issue.
This fix fixes 17210.
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Change `import readers.Dataset` to `import dataset_ops.Dataset`,
due to the changes in some other places.
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Fix library dependency issues in bazel
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add dependency to bazel rules
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add license to lib and pip package
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Remove unneeded changes in bazel
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Address review feedback
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Fix sanity check
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add zlib dependency and include path
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add copts in bazel to address the discrepancy in clang and gcc
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
|
|
PiperOrigin-RevId: 184897758
|