Searched refs:offset_or_window_dims (Results 1 – 1 of 1) sorted by relevance
804 absl::Span<const int64> offset_or_window_dims, in PassthroughOperandToGatherOutputOrScatterUpdate() argument824 int64 offset_dim = offset_or_window_dims[i - collapsed]; in PassthroughOperandToGatherOutputOrScatterUpdate()826 offset_dim < offset_or_window_dims[i - collapsed - 1]) { in PassthroughOperandToGatherOutputOrScatterUpdate()849 absl::Span<const int64> offset_or_window_dims, in PassthroughGatherOutputOrScatterUpdateToOperand() argument862 int64 offset_dim = offset_or_window_dims[i - collapsed]; in PassthroughGatherOutputOrScatterUpdateToOperand()869 offset_dim < offset_or_window_dims[i - collapsed - 1]) { in PassthroughGatherOutputOrScatterUpdateToOperand()