Commit 8a4d8496 authored by Gary Deer's avatar Gary Deer Committed by Sourabh Bajaj
Browse files

Feature Request: C++ gradient for LRN (#13987)

* WIP: added stub method and failing test

* test with default bias,alpha,beta,depth_radius passes

* Made changes based on code review

Swapped out LRN for LRNGrad
renamed to LRNGradHelper

* Fixed call to LRNGrad with proper namespace

* fixed function call to LRNGrad, all tests pass somehow

* fixed function call to LRNGrad, tests pass

* minor formatting changes based on clang-format suggestions

* formatting changes and updated test based on code review

* trimmed the tensor shape to prevent test timeout
parent f42fde42
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment