MCPcopy Create free account

hub / github.com/NVlabs/prismer / functions

Functions669 in github.com/NVlabs/prismer

Methodforward
(self, hidden_states: torch.Tensor)
model/modules/roberta.py:166
Methodforward
(self, hidden_states: torch.Tensor, input_tensor: torch.Tensor)
model/modules/roberta.py:179
Methodforward
(self, hidden_states: torch.Tensor, attention_mask: Optional[torch.FloatTensor] = None, mode='
model/modules/roberta.py:193
Methodforward
( self, hidden_states: torch.Tensor, attention_mask: Optional[torch.FloatT
model/modules/roberta.py:212
Methodforward
( self, input_ids: Optional[torch.Tensor] = None, attention_mask: Optional
model/modules/roberta.py:294
Methodforward
( self, input_ids: Optional[torch.LongTensor] = None, attention_mask: Opti
model/modules/roberta.py:358
Methodforward
(self, features, **kwargs)
model/modules/roberta.py:421
Methodforward
(self, x: torch.Tensor, latents: torch.Tensor)
model/modules/resampler.py:33
Methodforward
(self, x_f: torch.Tensor)
model/modules/resampler.py:46
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/model.py:58
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/model.py:85
Methodforward
(self, feat)
experts/ocr_detection/charnet/modeling/model.py:107
Methodforward
(self, im, im_scale_w, im_scale_h, original_im_w, original_im_h)
experts/ocr_detection/charnet/modeling/model.py:145
Methodforward
( self, pred_word_fg, pred_word_tblr, pred_word_orient, pred_char_fg, pred
experts/ocr_detection/charnet/modeling/postprocessing.py:66
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/backbone/decoder.py:37
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/backbone/resnet.py:136
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/backbone/resnet.py:188
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/backbone/resnet.py:291
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/backbone/resnet.py:326
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/backbone/hourglass.py:54
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/backbone/hourglass.py:75
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/backbone/hourglass.py:98
Methodforward
(ctx, x, new_shape)
experts/ocr_detection/charnet/modeling/layers/misc.py:19
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/layers/misc.py:46
Methodforward
(self, x)
experts/ocr_detection/charnet/modeling/layers/misc.py:67
Methodforward
(self, input)
experts/ocr_detection/charnet/modeling/layers/scale.py:17
Methodforward
(self, x)
experts/edge/model.py:44
Methodforward
(self, x)
experts/edge/model.py:66
Methodforward
(self, x)
experts/edge/model.py:112
Methodforward
(self, x)
experts/edge/model.py:126
Methodforward
(self, x)
experts/edge/model.py:150
Methodforward
(self, x)
experts/depth/vit.py:62
Methodforward
(self, x)
experts/depth/vit.py:71
Methodforward
(self, x)
experts/depth/vit.py:86
Methodforward
(self, x)
experts/depth/vit.py:99
Methodforward
Forward pass. Args: x (tensor): input Returns: tensor: interpolated data_list
experts/depth/blocks.py:155
Methodforward
Forward pass. Args: x (tensor): input Returns: tensor: output
experts/depth/blocks.py:196
Methodforward
Forward pass. Returns: tensor: output
experts/depth/blocks.py:227
Methodforward
Forward pass. Args: x (tensor): input Returns: tensor: output
experts/depth/blocks.py:290
Methodforward
Forward pass. Returns: tensor: output
experts/depth/blocks.py:362
Methodforward
(self, x)
experts/depth/models.py:114
Methodforward
Args: batched_inputs: a list, batched outputs of :class:`DatasetMapper`. Each item in the list contains the input
experts/segmentation/mask2former/maskformer_model.py:167
Methodforward
Performs the matching Params: outputs: This is a dict that contains at least these entries: "pred_logits": Tenso
experts/segmentation/mask2former/modeling/matcher.py:159
Methodforward
This performs the loss computation. Parameters: outputs: dict of tensors, see the output specification of the model for the forma
experts/segmentation/mask2former/modeling/criterion.py:212
Methodforward
(self, src, mask, query_embed, pos_embed)
experts/segmentation/mask2former/modeling/transformer_decoder/transformer.py:61
Methodforward
( self, src, mask: Optional[Tensor] = None, src_key_padding_mask: Optional[Ten
experts/segmentation/mask2former/modeling/transformer_decoder/transformer.py:85
Methodforward
( self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: O
experts/segmentation/mask2former/modeling/transformer_decoder/transformer.py:113
Methodforward
( self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional
experts/segmentation/mask2former/modeling/transformer_decoder/transformer.py:218
Methodforward
( self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: O
experts/segmentation/mask2former/modeling/transformer_decoder/transformer.py:323
Methodforward
(self, x, mask_features, mask=None)
experts/segmentation/mask2former/modeling/transformer_decoder/maskformer_transformer_decoder.py:130
Methodforward
(self, x)
experts/segmentation/mask2former/modeling/transformer_decoder/maskformer_transformer_decoder.py:185
Methodforward
(self, x, mask=None)
experts/segmentation/mask2former/modeling/transformer_decoder/position_encoding.py:29
Methodforward
(self, tgt, tgt_mask: Optional[Tensor] = None, tgt_key_padding_mask: Optional[
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:64
Methodforward
(self, tgt, memory, memory_mask: Optional[Tensor] = None, memory_key_padding_m
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:126
Methodforward
(self, tgt)
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:175
Methodforward
(self, x)
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:201
Methodforward
(self, x, mask_features, mask = None)
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:363
Methodforward
Forward function. Args: x: input features with shape of (num_windows*B, N, C) mask: (0/-inf) mask with shape of (num_w
experts/segmentation/mask2former/modeling/backbone/swin.py:131
Methodforward
Forward function. Args: x: Input feature, tensor size (B, H*W, C). H, W: Spatial resolution of the input feature.
experts/segmentation/mask2former/modeling/backbone/swin.py:235
Methodforward
Forward function. Args: x: Input feature, tensor size (B, H*W, C). H, W: Spatial resolution of the input feature.
experts/segmentation/mask2former/modeling/backbone/swin.py:311
Methodforward
Forward function. Args: x: Input feature, tensor size (B, H*W, C). H, W: Spatial resolution of the input feature.
experts/segmentation/mask2former/modeling/backbone/swin.py:406
Methodforward
Forward function.
experts/segmentation/mask2former/modeling/backbone/swin.py:479
Methodforward
Forward function.
experts/segmentation/mask2former/modeling/backbone/swin.py:651
Methodforward
Args: x: Tensor of shape (N,C,H,W). H, W must be a multiple of ``self.size_divisibility``. Returns: dict[str-
experts/segmentation/mask2former/modeling/backbone/swin.py:743
Methodforward
(self, features, mask=None)
experts/segmentation/mask2former/modeling/meta_arch/mask_former_head.py:115
Methodforward
Returns: In training, returns (None, dict of losses) In inference, returns (CxHxW logits, {})
experts/segmentation/mask2former/modeling/meta_arch/per_pixel_baseline.py:94
Methodforward
Returns: In training, returns (None, dict of losses) In inference, returns (CxHxW logits, {})
experts/segmentation/mask2former/modeling/meta_arch/per_pixel_baseline.py:208
Methodforward
(self, features, targets=None)
experts/segmentation/mask2former/modeling/pixel_decoder/fpn.py:156
Methodforward
(self, src, mask, pos_embed)
experts/segmentation/mask2former/modeling/pixel_decoder/fpn.py:191
Methodforward
(self, features, targets=None)
experts/segmentation/mask2former/modeling/pixel_decoder/fpn.py:309
Methodforward
(self, srcs, pos_embeds)
experts/segmentation/mask2former/modeling/pixel_decoder/msdeformattn.py:61
Methodforward
(self, src, pos, reference_points, spatial_shapes, level_start_index, padding_mask=None)
experts/segmentation/mask2former/modeling/pixel_decoder/msdeformattn.py:122
Methodforward
(self, src, spatial_shapes, level_start_index, valid_ratios, pos=None, padding_mask=None)
experts/segmentation/mask2former/modeling/pixel_decoder/msdeformattn.py:155
Methodforward
:param query (N, Length_{query}, C) :param reference_points (N, Length_{query}, n_levels, 2), range
experts/segmentation/mask2former/modeling/pixel_decoder/ops/modules/ms_deform_attn.py:82
Methodforward
(ctx, value, value_spatial_shapes, value_level_start_index, sampling_locations, attention_weights, im2col_step
experts/segmentation/mask2former/modeling/pixel_decoder/ops/functions/ms_deform_attn_func.py:34
Methodforward
(self, *args)
experts/normal/utils/losses.py:28
Methodforward
(self, img, **kwargs)
experts/normal/models/NNET.py:21
Methodforward
(self, x, **kwargs)
experts/normal/models/baseline.py:15
Methodforward
(self, x)
experts/normal/models/baseline.py:48
Methodforward
(self, features)
experts/normal/models/baseline.py:70
Methodforward
(self, features, gt_norm_mask=None, mode='test')
experts/normal/models/submodules/decoder.py:59
Methodforward
(self, x)
experts/normal/models/submodules/encoder.py:22
Methodforward
(self, x, concat_with)
experts/normal/models/submodules/submodules.py:21
Methodforward
(self, x, concat_with)
experts/normal/models/submodules/submodules.py:39
Methodforward
(self, x)
experts/normal/models/submodules/submodules.py:52
Methodforward
(self, x)
experts/obj_detection/unidet/modeling/backbone/resnest.py:107
Methodforward
(self, x)
experts/obj_detection/unidet/modeling/backbone/resnest.py:249
Methodforward
(self, x)
experts/obj_detection/unidet/modeling/backbone/resnest.py:412
Methodforward
(self, x)
experts/obj_detection/unidet/modeling/backbone/resnest.py:507
Methodforward
(self, x)
experts/obj_detection/unidet/modeling/backbone/resnest.py:585
Methodforward
(self, c5)
experts/obj_detection/unidet/modeling/backbone/resnest.py:785
Methodforward
(self, x)
experts/obj_detection/unidet/modeling/backbone/splat.py:61
Methodforward
(self, x)
experts/obj_detection/unidet/modeling/backbone/splat.py:102
Methodforward
(self, x, offset_input)
experts/obj_detection/unidet/modeling/backbone/splat.py:156
Methodforward
(self, c5)
experts/obj_detection/unidet/modeling/backbone/fpn_p5.py:30
Methodforward
(self, x, dataset_source=-1)
experts/obj_detection/unidet/modeling/roi_heads/multi_dataset_fast_rcnn.py:36
Methodforward
(self, images, features, proposals, targets=None, eval_dataset=-1)
experts/obj_detection/unidet/modeling/roi_heads/unified_roi_heads.py:44
Methodforward
(self, images, features, proposals, targets=None, eval_dataset=-1)
experts/obj_detection/unidet/modeling/roi_heads/split_roi_heads.py:64
Methodforward
(self, batched_inputs)
experts/obj_detection/unidet/modeling/meta_arch/split_rcnn.py:28
Methodforward
(self, batched_inputs)
experts/obj_detection/unidet/modeling/meta_arch/unified_rcnn.py:37
← previousnext →501–600 of 669, ranked by callers