Searched refs:CalculateGradWarp (Results 1 – 1 of 1) sorted by relevance
359 XlaOp CalculateGradWarp(XlaOpKernelContext* ctx, XlaOp grad_output, XlaOp ratio, in CalculateGradWarp() function663 CalculateGradWarp(ctx, grad_output, ratio, gather_indices, data, in Compile()