Searched defs:GradOp (Results 1 – 1 of 1) sorted by relevance
71 struct GradOp { struct72 __device__ GradOp(accscalar_t m, const PTA& i, const PTA& g) in GradOp() function74 __device__ __forceinline__ Float2<scalar_t, accscalar_t> operator()(int batch, int plane, int n) { in operator ()()79 const accscalar_t mean;80 const PTA& input;81 const PTA& grad_output;