fix gradient based criterion bug

[wenhuach21]

Signed-off-by: wenhuach21 [email protected]

Type of Change

feature or bug fix or documentation or validation or others
API changed or not

Description

detail description JIRA ticket: xxx

Expected Behavior & Potential Risk

the expected behavior that triggered by this PR

How has this PR been tested?

how to reproduce the test (including hardware information)

Dependency Change?

any library dependency introduced or removed