Searched refs:offset_dim (Results 1 – 2 of 2) sorted by relevance
824 int64 offset_dim = offset_or_window_dims[i - collapsed]; in PassthroughOperandToGatherOutputOrScatterUpdate() local826 offset_dim < offset_or_window_dims[i - collapsed - 1]) { in PassthroughOperandToGatherOutputOrScatterUpdate()830 passthrough_tile[offset_dim] = dim_partitions; in PassthroughOperandToGatherOutputOrScatterUpdate()862 int64 offset_dim = offset_or_window_dims[i - collapsed]; in PassthroughGatherOutputOrScatterUpdateToOperand() local864 update_or_gather_sharding.tile_assignment().dim(offset_dim); in PassthroughGatherOutputOrScatterUpdateToOperand()869 offset_dim < offset_or_window_dims[i - collapsed - 1]) { in PassthroughGatherOutputOrScatterUpdateToOperand()
3232 int64 offset_dim = dim_numbers.offset_dims(i); in ValidateGatherDimensionNumbers() local3233 if (offset_dim < 0 || offset_dim >= output_shape_rank) { in ValidateGatherDimensionNumbers()3238 i, offset_dim, output_shape_rank); in ValidateGatherDimensionNumbers()