Small optimization for tfa.image.transform_ops.angles_to_projective_transforms

[Smankusors]

Description

Brief Description of the PR:

Fixes #2054.

Type of change

• Bug fix
• New Tutorial
• Updated or additional documentation
• Additional Testing
• New Activation and the changes conform to the activation contribution guidelines
• New Callback and the changes conform to the callback contribution guidelines
• New Image addition and the changes conform to the image op contribution guidelines
• New Layer and the changes conform to the layer contribution guidelines
• New Loss and the changes conform to the loss contribution guidelines
• New Metric and the changes conform to the metric contribution guidelines
• New Optimizer and the changes conform to the optimizer contribution guidelines
• New RNN Cell and the changes conform to the rnn contribution guidelines
• New Seq2seq addition and the changes conform to the seq2seq contribution guidelines
• New Text addition and the changes conform to the text op contribution guidelines

Checklist:

• I've properly formatted my code according to the guidelines
  • By running Black + Flake8
  • By running pre-commit hooks
• This PR addresses an already submitted issue for TensorFlow Addons
• I have made corresponding changes to the documentation
• I have added tests that prove my fix is effective or that my feature works
• This PR contains modifications to C++ custom-ops

How Has This Been Tested?

I tested this on Google Colab environment. Well here's the result :

# CPU (Run 1)
angles_to_projective_transforms     0065868
angles_to_projective_transforms2    0069990
angles_to_projective_transforms_tf  0257738
angles_to_projective_transforms2_tf 0260291
# CPU (Run 2)
angles_to_projective_transforms     0065672
angles_to_projective_transforms2    0069954
angles_to_projective_transforms_tf  0261689
angles_to_projective_transforms2_tf 0254197
# CPU (Run 3)
angles_to_projective_transforms     0066451
angles_to_projective_transforms2    0070442
angles_to_projective_transforms_tf  0264665
angles_to_projective_transforms2_tf 0262863
# CPU (Run 4)
angles_to_projective_transforms     0071835
angles_to_projective_transforms2    0076597
angles_to_projective_transforms_tf  0294059
angles_to_projective_transforms2_tf 0286251

# GPU
angles_to_projective_transforms     0029724
angles_to_projective_transforms2    0033601
angles_to_projective_transforms_tf  0230617
angles_to_projective_transforms2_tf 0230509

# TPU
angles_to_projective_transforms     0058424
angles_to_projective_transforms2    0061462
angles_to_projective_transforms_tf  0293018
angles_to_projective_transforms2_tf 0302889

(The higher the number, the better)

Description about each row :

• angles_to_projective_transforms : is the original function
• angles_to_projective_transforms2 : is after the small optimization
• angles_to_projective_transforms_tf : is the original function with tf.function decoration
• angles_to_projective_transforms2_tf : is after the small optimization with tf.function decoration

Calling it directly seems have a performance increase. But if decorated with tf.function, it seems like it's already optimized by Tensorflow, so not really seeing performance increase there.

The full benchmark script is here

[googlebot]

Thanks for your pull request. It looks like this may be your first contribution to a Google open source project (if not, look below for help). Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

📝 Please visit https://cla.developers.google.com/ to sign.

Once you've signed (or fixed any issues), please reply here with @googlebot I signed it! and we'll verify it.

---

What to do if you already signed the CLA

Individual signers

• It's possible we don't have your GitHub username or you're using a different email address on your commit. Check your existing CLA data and verify that your email is set on your git commits.

Corporate signers

• Your company has a Point of Contact who decides which employees are authorized to participate. Ask your POC to be added to the group of authorized contributors. If you don't know who your Point of Contact is, direct the Google project maintainer to go/cla#troubleshoot (Public version).
• The email used to register you as an authorized contributor must be the email used for the Git commit. Check your existing CLA data and verify that your email is set on your git commits.
• The email used to register you as an authorized contributor must also be attached to your GitHub account.

ℹ️ Googlers: Go here for more info.

[bot-of-gabrieldemarmiesse]

@mels630

You are owner of some files modified in this pull request.
Would you kindly review the changes whenever you have the time to?
Thank you very much.

[Smankusors]

@googlebot I signed it!

[googlebot]

CLAs look good, thanks!

ℹ️ Googlers: Go here for more info.

[WindQAQ]

Thanks :-)