Schema Exploration...

xs:element/*[@name='test1a' or @ref='test1b']
xs:element/xs:complexType/xs:choice/*[@name='test2a' or @ref='test2b']
xs:element/xs:complexType/xs:sequence/*[@name='test3a' or @ref='test3b']

...
  <xs:element name="test1">
     <xs:element name="test1a"/>
     <xs:element ref="test1b"/>
  </xs:element>

  <xs:element name="test2">
     <xs:complexType>
         <xs:choice>
            <xs:element name="test2a"/>
            <xs:element ref="test2b"/>
            ...
         </xs:choice>
      </xs:complexType>
  </xs:element>

  <xs:element name="test3">
     <xs:complexType>
	<xs:sequence>
            <xs:element name="test3a"/>
            <xs:element ref="test3b"/>
             ...
         </xs:sequence>
      </xs:complexType>
  </xs:element>
...

<xsl:apply-templates select="
$schema_node/xs:element[@name=$node_name or @ref=$node_name]
"/>