Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refactor TF Dataset code and lint it #2909

Merged
merged 3 commits into from
Apr 29, 2021
Merged

Conversation

klecki
Copy link
Contributor

@klecki klecki commented Apr 27, 2021

Extract the dataset and iterator classes
out of being nested in DALIDatasetOp kernel.

Enable linting in dali_tf_plugin and apply fixes.

Signed-off-by: Krzysztof Lecki klecki@nvidia.com

Why we need this PR?

  • Refactoring to improve TF DALI Dataset.

What happened in this PR?

  • What solution was applied:
    Linter was turend on,
    Code was un-nested and dataset split into .h and .cc file
    Code was reformatted.
  • Affected modules and functionalities:
    TF Dataset <- only source code rearrangement.
    Only linter and build scripts contain meaningful changes.
  • Key points relevant for the review:
    NA
  • Validation and testing:
    CI
  • Documentation (including examples):
    Not really.

JIRA TASK: [Use DALI-XXXX or NA]

Extract the dataset and iterator classes
out of being nested in DALIDatasetOp kernel

Signed-off-by: Krzysztof Lecki <klecki@nvidia.com>
Signed-off-by: Krzysztof Lecki <klecki@nvidia.com>
@klecki
Copy link
Contributor Author

klecki commented Apr 27, 2021

!build

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [2310348]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [2310348]: BUILD FAILED

Signed-off-by: Krzysztof Lecki <klecki@nvidia.com>
@dali-automaton
Copy link
Collaborator

CI MESSAGE: [2313371]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [2313371]: BUILD PASSED

@klecki klecki merged commit dccf77d into NVIDIA:master Apr 29, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants