Searched refs:shifting_value (Results 1 – 1 of 1) sorted by relevance
312 auto shifting_value = xla::ConstantR1<int32>( in CalculateGradData() local315 xla::Add(gather_indices, shifting_value, {last_warp_dim}); in CalculateGradData()385 auto shifting_value = xla::ConstantR1<int32>( in CalculateGradWarp() local388 xla::Add(gather_indices, shifting_value, {last_warp_dim}); in CalculateGradWarp()