updated gradient test input to only include the input array; updated baseline too
Authored by: obrientaiu 2022-05-18
Parent: [r138]
Child: [r140]