We should use `INPUT_CHANNEL_ATTRIBUTE` here; will fix during merge.