Skip to content
This repository was archived by the owner on Apr 19, 2026. It is now read-only.
This repository was archived by the owner on Apr 19, 2026. It is now read-only.

Performance issue in the definition of _sliced_wasserstein, tensorflow_gan/python/eval/sliced_wasserstein.py #49

@DLPerf

Description

@DLPerf

Hello, I found a performance issue in the definition of _sliced_wasserstein, tensorflow_gan/python/eval/sliced_wasserstein.py, tf.shape(input=a)[1] will calculated repeatedly during the program execution, resulting in reduced efficiency. I think it should be created before the loop in the _sliced_wasserstein.

Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions