def _add_pooler_output()

in src/exporters/coreml/config.py [0:0]

• 11 lines of code
• 3 McCabe index (conditional complexity)

    def _add_pooler_output(self, output_descs):
        if self.task == "feature-extraction":
            if self.modality == "vision":
                description = "Last layer hidden-state after a pooling operation on the spatial dimensions"
            else:
                description = "Last layer hidden-state of the first token of the sequence"

            output_descs["pooler_output"] = OutputDescription(
                "pooler_output",
                description
            )
        return output_descs