Methodbackward input: grad_out: (n, c) output: grad_input: (n, c), grad_position: (n, nsample, c), grad_weight : (n, nsample, c')
libs/pointops/functions/aggregation.py:22
Methodbackward input: grad_out: (n, nsample, c) output: grad_input1: (n, c), grad_input2: (n, c)
libs/pointops/functions/subtraction.py:22
Methodbackward input: coords: (m, 3), new_xyz: (n, 3), input: (m, c), offset: (b), new_offset: (b) output: (n, c)
libs/pointops/functions/interpolation.py:46
Methodbackward input: grad_output: (n, g, c) output: grad_weight: (m, g), grad_value: (n, g, c), none, none
libs/pointops/functions/attention.py:75
Methodforward input: input: (n, c), position: (n, nsample, c), weight : (n, nsample, c'), idx: (n, nsample) output: (n, c)
libs/pointops/functions/aggregation.py:9
Methodforward input: input: (n, c), idx : (m, nsample) output: (m, nsample, c)
libs/pointops/functions/grouping.py:9
Methodforward input: input1: (n, c), input2: (n, c), idx: (n, nsample) output: (n, nsample, c)
libs/pointops/functions/subtraction.py:9
Methodforward input: coords: (m, 3), new_xyz: (n, 3), input: (m, c), offset: (b), new_offset: (b) output: (n, c)
libs/pointops/functions/interpolation.py:27
Methodforward input - query: (n, g, c), key: (n, g, c), weight: (c) 1_c for scatter attention, index_target: (m), index_refer: (m)
libs/pointops/functions/attention.py:10
Methodforward input - weight: (m, g), value: (n, g, c) index_target: (m), index_value: (m) output - output: (n, g, c)
libs/pointops/functions/attention.py:52
Methodforward input: coords: (n, 3), new_xyz: (m, 3), offset: (b), new_offset: (b) output: idx: (m, nsample) -1 is placeholder, dist2: (m, nsample)
libs/pointops/functions/query.py:9
Methodforward input: coords: (n, 3), new_xyz: (m, 3), offset: (b), new_offset: (b) output: idx: (m, nsample), dist2: (m, nsample)
libs/pointops/functions/query.py:32
Methodforward input: coords: (n, 3), new_xyz: (m, 3), offset: (b), new_offset: (b) output: idx: (m, nsample), dist2: (m, nsample)
libs/pointops/functions/query.py:63
Functionhandle_process(scene_path, output_path, labels_pd, train_scenes, val_scenes, parse_normals=True)
scannet-preprocess/preprocess_scannet.py:97
Functioninterpolation input: coords: (m, 3), new_xyz: (n, 3), color: (m, c), offset: (b), new_offset: (b) output: (n, c)
libs/pointops/functions/interpolation.py:8
Functionquery_and_group input: coords: (n, 3), new_xyz: (m, 3), color: (n, c), idx: (m, nsample), offset: (b), new_offset: (b) output: new_feat: (m, nsample, c+3), g
libs/pointops/functions/utils.py:38