↓ 3 callersFunctioncorrelation(input1, input2, pad_size, kernel_size, max_displacement, stride1, stride2, corr_type_multiply=1)
CV/PWCNet/correlation_op/correlation.py:21
↓ 3 callersFunctiondot_product_attention Args: query: a tensor with shape [batch, Q_time, Q_dimension] key: a tensor with shape [batch, time, K_dimension] val
ST_DM/KDD2021-MSTPAC/code/MST-PAC/nets/poi_qac_personalized/qac_personalized.py:132
↓ 3 callersMethodencode(self, text_input=None, text_mask=None, gather_idx=None,
image_input=None, image_mask=None, dec
NLP/UNIMO-2/src/model/unimo_grounded.py:251
↓ 3 callersMethodencode(self, text_input=None, text_mask=None, gather_idx=None,
image_input=None, image_mask=None, dec
NLP/UNIMO-2/src/model/unimo_grounded_baseline.py:215
↓ 3 callersMethodencode(self, text_input=None, text_mask=None, gather_idx=None,
image_input=None, image_mask=None, dec
NLP/UNIMO-2/src/finetune/img2txt_oscar.py:781