↓ 4 callersMethodforward(
self, model, x_lhand, x_rhand,
x_obj, obj_feat,
timesteps=None, enc_text=None,
Affordance-DrivenHOIDiffusion/lib/networks/diffusion copy.py:78
↓ 4 callersFunctionproc_obj_feat_final(
contact_estimator,
obj_scale, obj_cent,
obj_feat, enc_text, npts,
use_obj_scale_centr
Affordance-DrivenHOIDiffusion/lib/utils/proc.py:321
↓ 3 callersFunction_flash_attn_backward(do, q, k, v, o, lse, dq, dk, dv, bias=None, causal=False, softmax_scale=None)
HOIAffordanceMLLM/llava/model/language_model/mpt/flash_attn_triton.py:366
↓ 3 callersFunction_flash_attn_forward(q, k, v, bias=None, causal=False, softmax_scale=None)
HOIAffordanceMLLM/llava/model/language_model/mpt/flash_attn_triton.py:329
↓ 3 callersFunctionfarthest_point_sample Input: xyz: pointcloud data, [B, N, 3] npoint: number of samples Return: centroids: sampled pointcloud index, [B, npo
HOIAffordanceMLLM/ReConV2/segmentation/pointnet_util.py:53
↓ 3 callersFunctionget_contact_info(
lhand_pose_list, lhand_beta_list, lhand_trans_list,
rhand_pose_list, rhand_beta_list, rha
Affordance-DrivenHOIDiffusion/lib/utils/proc.py:99
↓ 3 callersFunctionget_l2_loss(
pred_lhand=None, pred_rhand=None, pred_obj=None,
targ_lhand=None, targ_rhand=None, targ_obj
Affordance-DrivenHOIDiffusion/lib/utils/loss.py:24
↓ 3 callersFunctionget_object_hand_info(
object_model,
clip_model, text,
obj_root, data_config,
mpnet=None,
)
Affordance-DrivenHOIDiffusion/lib/utils/demo_utils.py:144
↓ 3 callersFunctionproc_cond_contact_estimator_cov_map(obj_scale, obj_feat, enc_text, npts, use_scale,affordance_map)
Affordance-DrivenHOIDiffusion/lib/utils/proc.py:247
↓ 2 callersMethod__init__(self, npoint, radius, nsample, in_channel, mlp, group_all, knn=False)
HOIAffordanceMLLM/ReConV2/segmentation/pointnet_util.py:160
↓ 2 callersMethod__init__(self, npoint, radius, nsample, in_channel, mlp, group_all)
HOIAffordanceMLLM/ReConV2/segmentation/models/pointnet2_utils.py:159
↓ 2 callersMethod__init__(self, normalized_shape, eps=1e-05, weight=True, dtype=None, device=None)
HOIAffordanceMLLM/llava/model/language_model/mpt/norm.py:35
↓ 2 callersMethod__init__(
self,
data_path,
data_obj_pc_path,
text_json,
max_nframes,
Affordance-DrivenHOIDiffusion/lib/datasets/grab.py:175
↓ 2 callersMethod__init__(
self,
data_path,
data_obj_pc_path,
text_json,
max_nframes,
Affordance-DrivenHOIDiffusion/lib/datasets/h2o.py:170
↓ 2 callersMethod__init__(
self,
data_path,
data_obj_pc_path,
text_json,
max_nframes,
Affordance-DrivenHOIDiffusion/lib/datasets/arctic.py:172
↓ 2 callersFunction_bwd_kernel_one_col_block(start_n, Q, K, V, Bias, DO, DQ, DK, DV, LSE, D, softmax_scale, stride_qm, stride_kn, stride_vn, stride_bm, st
HOIAffordanceMLLM/llava/model/language_model/mpt/flash_attn_triton.py:184
↓ 2 callersFunction_bwd_store_dk_dv(dk_ptrs, dv_ptrs, dk, dv, offs_n, offs_d, seqlen_k, headdim, EVEN_M: tl.constexpr, EVEN_N: tl.constexpr, EVEN
HOIAffordanceMLLM/llava/model/language_model/mpt/flash_attn_triton.py:168
↓ 2 callersFunction_normal_param_init_fn_(module: nn.Module, std: float, n_layers: int, d_model: Optional[int]=None, init_div_is_residual: Union[int, f
HOIAffordanceMLLM/llava/model/language_model/mpt/param_init_fns.py:124