Searched defs:x_coeffs (Results 1 – 5 of 5) sorted by relevance
249 XlaOp x_coeffs = ZerosLike(rhs); in TridiagonalSolverImpl() local337 auto x_coeffs = values[0]; in TridiagonalSolverImpl() local
154 opmath_t x_coeffs[4]; in upsample_bicubic2d_backward_out_frame() local
870 scalar_t x_coeffs[4]; in _grid_sampler_2d_cpu_fallback_backward() local
142 accscalar_t x_coeffs[4]; in upsample_bicubic2d_backward_out_frame() local
464 scalar_t x_coeffs[4]; in grid_sampler_2d_backward_kernel() local