Searched defs:grad_data (Results 1 – 2 of 2) sorted by relevance
167 XlaOp ScatterToGradData(XlaOpKernelContext* ctx, XlaOp grad_data, XlaOp indices, in ScatterToGradData()301 auto grad_data = xla::ConstantLiteral( in CalculateGradData() local658 auto grad_data = CalculateGradData( in Compile() local
146 const T* __restrict__ grad_output, T* __restrict__ grad_data, in ResamplerGrad2DKernel()