Methodforward input: xyz: (n, 3), new_xyz: (m, 3), offset: (b), new_offset: (b) output: idx: (m, nsample), dist2: (m, nsample)
pointops2/functions/pointops_ablation.py:32
Methodforward input: input: (n, c), idx : (m, nsample) output: (m, nsample, c)
pointops2/functions/pointops_ablation.py:50
Methodforward input: input1: (n, c), input2: (n, c), idx: (n, nsample) output: (n, nsample, c)
pointops2/functions/pointops_ablation.py:106
Methodforward input: input: (n, c), position: (n, nsample, c), weight : (n, nsample, c'), idx: (n, nsample) output: (n, c)
pointops2/functions/pointops_ablation.py:136
Methodforward input: xyz: (m, 3), new_xyz: (n, 3), input: (m, c), offset: (b), new_offset: (b) output: (n, c)
pointops2/functions/pointops_ablation.py:184
Methodforward input: xyz: (n, 3), new_xyz: (m, 3), offset: (b), new_offset: (b) output: idx: (m, nsample), dist2: (m, nsample)
pointops2/functions/pointops2.py:32
Methodforward input: input1: (n, c), input2: (n, c), idx: (n, nsample) output: (n, nsample, c)
pointops2/functions/pointops2.py:105
Methodforward input: input: (n, c), position: (n, nsample, c), weight : (n, nsample, c'), idx: (n, nsample) output: (n, c)
pointops2/functions/pointops2.py:135
Methodforward input: xyz: (m, 3), new_xyz: (n, 3), input: (m, c), offset: (b), new_offset: (b) output: (n, c)
pointops2/functions/pointops2.py:183
Methodforward input: xyz: (n, 3), new_xyz: (m, 3), offset: (b), new_offset: (b) output: idx: (m, nsample), dist2: (m, nsample)
pointops2/functions/pointops.py:36
Methodforward input: q: (N, h, C//h), k: (N, h, C//h), index0: (M), index1: (M) output: output: [N, h, C//h]
pointops2/functions/pointops.py:84
Methodforward input: q: (N, h, C//h), k: (N, h, C//h), index0: (M), index1: (M) output: output: [N, h, C//h]
pointops2/functions/pointops.py:144
Methodforward input: attn: (M, h), v: (N, h, C//h), index0: (M), index1: (M) output: output: [N, h, C//h]
pointops2/functions/pointops.py:209
Methodforward input: attn: (M, h), v: (N, h, C//h), index0: (M), index1: (M) output: output: [L, h, C//h]
pointops2/functions/pointops.py:270
Methodforward input: q: (N, h, hdim), index: (M), table: (L, h, hdim, 3), rel_idx: (M, 3) output: output: [M, h]
pointops2/functions/pointops.py:322
Methodforward input: q: (N, h, hdim), index_q: (M), k: (N, h, hdim), index_k: (M), table_q: (L, h, hdim, 3), table_k: (L, h, hdim, 3), rel_idx: (M, 3)
pointops2/functions/pointops.py:374
Methodforward input: q: (N, h, hdim), index_q: (M), k: (N, h, hdim), index_k: (M), table_q: (L, h, hdim, 3), table_k: (L, h, hdim, 3), rel_idx: (M, 3)
pointops2/functions/pointops.py:448
Methodforward input: attn: (M, h), v: (N, h, hdim), index0: (M), index1: (M), table: (L, h, hdim, 3), rel_idx: (M, 3) output: output: [N, h, hdim]
pointops2/functions/pointops.py:523
Methodforward input: attn: (M, h), v: (N, h, hdim), index0_offsets: (M), index1: (M), table: (L, h, hdim, 3), rel_idx: (M, 3) output: output: [N, h
pointops2/functions/pointops.py:586
Methodforward input: input1: (n, c), input2: (n, c), idx: (n, nsample) output: (n, nsample, c)
pointops2/functions/pointops.py:697
Methodforward input: input: (n, c), position: (n, nsample, c), weight : (n, nsample, c'), idx: (n, nsample) output: (n, c)
pointops2/functions/pointops.py:727
Methodforward input: xyz: (m, 3), new_xyz: (n, 3), input: (m, c), offset: (b), new_offset: (b) output: (n, c)
pointops2/functions/pointops.py:802
Methodforward(
ctx,
means3D,
means2D,
sh,
colors_precomp,
flow_2d,
gaussian_renderer/diff_gaussian_rasterization.py:67
Methodforward(self, means3D, means2D, opacities, shs = None, colors_precomp = None, flow_2d = None, ts=None,
gaussian_renderer/diff_gaussian_rasterization.py:255
Functioninterpolation input: xyz: (m, 3), new_xyz: (n, 3), feat: (m, c), offset: (b), new_offset: (b) output: (n, c)
pointops2/functions/pointops_ablation.py:165
Functioninterpolation input: xyz: (m, 3), new_xyz: (n, 3), feat: (m, c), offset: (b), new_offset: (b) output: (n, c)
pointops2/functions/pointops2.py:164
Functioninterpolation input: xyz: (m, 3), new_xyz: (n, 3), feat: (m, c), offset: (b), new_offset: (b) output: (n, c)
pointops2/functions/pointops.py:756
Functioninterpolation_v2 input: xyz: (m, 3), new_xyz: (n, 3), feat: (m, c), offset: (b), new_offset: (b) output: (n, c)
pointops2/functions/pointops.py:773
Functionqueryandgroup input: xyz: (n, 3), new_xyz: (m, 3), feat: (n, c), idx: (m, nsample), offset: (b), new_offset: (b) output: new_feat: (m, c+3, nsample), group
pointops2/functions/pointops_ablation.py:79
Functionqueryandgroup input: xyz: (n, 3), new_xyz: (m, 3), feat: (n, c), idx: (m, nsample), offset: (b), new_offset: (b) output: new_feat: (m, c+3, nsample), group
pointops2/functions/pointops2.py:79
Functionqueryandgroup input: xyz: (n, 3), new_xyz: (m, 3), feat: (n, c), idx: (m, nsample), offset: (b), new_offset: (b) output: new_feat: (m, c+3, nsample), group
pointops2/functions/pointops.py:648
FunctionreadNerfSyntheticInfo(path, white_background, eval, extension=".png", num_pts=100_000, time_duration=None, num_extra_pts=0, frame_r
scene/dataset_readers.py:310