Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
one
spconv
Commits
82fd7a8b
Commit
82fd7a8b
authored
Nov 10, 2021
by
yan.yan
Browse files
v2.1.5: add profile tool and python 3.6 for linux
parent
f31eee3a
Changes
80
Show whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
97 additions
and
452 deletions
+97
-452
spconv/core_cc/csrc/sparse/__init__.pyi
spconv/core_cc/csrc/sparse/__init__.pyi
+0
-14
spconv/core_cc/csrc/sparse/all/ops1d.pyi
spconv/core_cc/csrc/sparse/all/ops1d.pyi
+22
-1
spconv/core_cc/csrc/sparse/all/ops1d/__init__.pyi
spconv/core_cc/csrc/sparse/all/ops1d/__init__.pyi
+0
-49
spconv/core_cc/csrc/sparse/all/ops1d/p2v_c.pyi
spconv/core_cc/csrc/sparse/all/ops1d/p2v_c.pyi
+0
-11
spconv/core_cc/csrc/sparse/all/ops2d.pyi
spconv/core_cc/csrc/sparse/all/ops2d.pyi
+22
-1
spconv/core_cc/csrc/sparse/all/ops2d/__init__.pyi
spconv/core_cc/csrc/sparse/all/ops2d/__init__.pyi
+0
-49
spconv/core_cc/csrc/sparse/all/ops2d/p2v_c.pyi
spconv/core_cc/csrc/sparse/all/ops2d/p2v_c.pyi
+0
-11
spconv/core_cc/csrc/sparse/all/ops3d.pyi
spconv/core_cc/csrc/sparse/all/ops3d.pyi
+22
-1
spconv/core_cc/csrc/sparse/all/ops3d/__init__.pyi
spconv/core_cc/csrc/sparse/all/ops3d/__init__.pyi
+0
-49
spconv/core_cc/csrc/sparse/all/ops3d/p2v_c.pyi
spconv/core_cc/csrc/sparse/all/ops3d/p2v_c.pyi
+0
-11
spconv/core_cc/csrc/sparse/all/ops4d.pyi
spconv/core_cc/csrc/sparse/all/ops4d.pyi
+22
-1
spconv/core_cc/csrc/sparse/all/ops4d/__init__.pyi
spconv/core_cc/csrc/sparse/all/ops4d/__init__.pyi
+0
-49
spconv/core_cc/csrc/sparse/all/ops4d/p2v_c.pyi
spconv/core_cc/csrc/sparse/all/ops4d/p2v_c.pyi
+0
-11
spconv/core_cc/csrc/sparse/all/ops_cpu1d.pyi
spconv/core_cc/csrc/sparse/all/ops_cpu1d.pyi
+3
-8
spconv/core_cc/csrc/sparse/all/ops_cpu1d/__init__.pyi
spconv/core_cc/csrc/sparse/all/ops_cpu1d/__init__.pyi
+0
-74
spconv/core_cc/csrc/sparse/all/ops_cpu1d/p2v_c.pyi
spconv/core_cc/csrc/sparse/all/ops_cpu1d/p2v_c.pyi
+0
-11
spconv/core_cc/csrc/sparse/all/ops_cpu2d.pyi
spconv/core_cc/csrc/sparse/all/ops_cpu2d.pyi
+3
-8
spconv/core_cc/csrc/sparse/all/ops_cpu2d/__init__.pyi
spconv/core_cc/csrc/sparse/all/ops_cpu2d/__init__.pyi
+0
-74
spconv/core_cc/csrc/sparse/all/ops_cpu2d/p2v_c.pyi
spconv/core_cc/csrc/sparse/all/ops_cpu2d/p2v_c.pyi
+0
-11
spconv/core_cc/csrc/sparse/all/ops_cpu3d.pyi
spconv/core_cc/csrc/sparse/all/ops_cpu3d.pyi
+3
-8
No files found.
spconv/core_cc/csrc/sparse/__init__.pyi
View file @
82fd7a8b
# Copyright 2021 Yan Yan
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
spconv/core_cc/csrc/sparse/all/ops1d.pyi
View file @
82fd7a8b
...
...
@@ -19,10 +19,31 @@ class Point2Voxel:
max_num_points_per_voxel:
"""
...
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True
, empty_mean: bool = False, stream_int: int = 0
) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
empty_mean:
stream_int:
"""
...
@staticmethod
def point_to_voxel_hash_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, hashdata: Tensor, point_indice_data: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
hashdata:
point_indice_data:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
empty_mean:
stream_int:
"""
...
spconv/core_cc/csrc/sparse/all/ops1d/__init__.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
from cumm.tensorview import Tensor
class Point2Voxel:
hashdata: Tensor
point_indice_data: Tensor
voxels: Tensor
indices: Tensor
num_per_voxel: Tensor
@property
def grid_size(self) -> List[int]: ...
def __init__(self, vsize_xyz: List[float], coors_range_xyz: List[float], num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int) -> None:
"""
Args:
vsize_xyz:
coors_range_xyz:
num_point_features:
max_num_voxels:
max_num_points_per_voxel:
"""
...
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
empty_mean:
stream_int:
"""
...
@staticmethod
def point_to_voxel_hash_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, hashdata: Tensor, point_indice_data: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
hashdata:
point_indice_data:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
empty_mean:
stream_int:
"""
...
spconv/core_cc/csrc/sparse/all/ops1d/p2v_c.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
class Point2VoxelCommon:
@staticmethod
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]) -> Tuple[List[float], List[int], List[int], List[float]]:
"""
Args:
vsize_xyz:
coors_range_xyz:
"""
...
spconv/core_cc/csrc/sparse/all/ops2d.pyi
View file @
82fd7a8b
...
...
@@ -19,10 +19,31 @@ class Point2Voxel:
max_num_points_per_voxel:
"""
...
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True
, empty_mean: bool = False, stream_int: int = 0
) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
empty_mean:
stream_int:
"""
...
@staticmethod
def point_to_voxel_hash_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, hashdata: Tensor, point_indice_data: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
hashdata:
point_indice_data:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
empty_mean:
stream_int:
"""
...
spconv/core_cc/csrc/sparse/all/ops2d/__init__.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
from cumm.tensorview import Tensor
class Point2Voxel:
hashdata: Tensor
point_indice_data: Tensor
voxels: Tensor
indices: Tensor
num_per_voxel: Tensor
@property
def grid_size(self) -> List[int]: ...
def __init__(self, vsize_xyz: List[float], coors_range_xyz: List[float], num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int) -> None:
"""
Args:
vsize_xyz:
coors_range_xyz:
num_point_features:
max_num_voxels:
max_num_points_per_voxel:
"""
...
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
empty_mean:
stream_int:
"""
...
@staticmethod
def point_to_voxel_hash_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, hashdata: Tensor, point_indice_data: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
hashdata:
point_indice_data:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
empty_mean:
stream_int:
"""
...
spconv/core_cc/csrc/sparse/all/ops2d/p2v_c.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
class Point2VoxelCommon:
@staticmethod
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]) -> Tuple[List[float], List[int], List[int], List[float]]:
"""
Args:
vsize_xyz:
coors_range_xyz:
"""
...
spconv/core_cc/csrc/sparse/all/ops3d.pyi
View file @
82fd7a8b
...
...
@@ -19,10 +19,31 @@ class Point2Voxel:
max_num_points_per_voxel:
"""
...
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True
, empty_mean: bool = False, stream_int: int = 0
) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
empty_mean:
stream_int:
"""
...
@staticmethod
def point_to_voxel_hash_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, hashdata: Tensor, point_indice_data: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
hashdata:
point_indice_data:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
empty_mean:
stream_int:
"""
...
spconv/core_cc/csrc/sparse/all/ops3d/__init__.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
from cumm.tensorview import Tensor
class Point2Voxel:
hashdata: Tensor
point_indice_data: Tensor
voxels: Tensor
indices: Tensor
num_per_voxel: Tensor
@property
def grid_size(self) -> List[int]: ...
def __init__(self, vsize_xyz: List[float], coors_range_xyz: List[float], num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int) -> None:
"""
Args:
vsize_xyz:
coors_range_xyz:
num_point_features:
max_num_voxels:
max_num_points_per_voxel:
"""
...
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
empty_mean:
stream_int:
"""
...
@staticmethod
def point_to_voxel_hash_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, hashdata: Tensor, point_indice_data: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
hashdata:
point_indice_data:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
empty_mean:
stream_int:
"""
...
spconv/core_cc/csrc/sparse/all/ops3d/p2v_c.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
class Point2VoxelCommon:
@staticmethod
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]) -> Tuple[List[float], List[int], List[int], List[float]]:
"""
Args:
vsize_xyz:
coors_range_xyz:
"""
...
spconv/core_cc/csrc/sparse/all/ops4d.pyi
View file @
82fd7a8b
...
...
@@ -19,10 +19,31 @@ class Point2Voxel:
max_num_points_per_voxel:
"""
...
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True
, empty_mean: bool = False, stream_int: int = 0
) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
empty_mean:
stream_int:
"""
...
@staticmethod
def point_to_voxel_hash_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, hashdata: Tensor, point_indice_data: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
hashdata:
point_indice_data:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
empty_mean:
stream_int:
"""
...
spconv/core_cc/csrc/sparse/all/ops4d/__init__.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
from cumm.tensorview import Tensor
class Point2Voxel:
hashdata: Tensor
point_indice_data: Tensor
voxels: Tensor
indices: Tensor
num_per_voxel: Tensor
@property
def grid_size(self) -> List[int]: ...
def __init__(self, vsize_xyz: List[float], coors_range_xyz: List[float], num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int) -> None:
"""
Args:
vsize_xyz:
coors_range_xyz:
num_point_features:
max_num_voxels:
max_num_points_per_voxel:
"""
...
def point_to_voxel_hash(self, points: Tensor, clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
empty_mean:
stream_int:
"""
...
@staticmethod
def point_to_voxel_hash_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, hashdata: Tensor, point_indice_data: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True, empty_mean: bool = False, stream_int: int = 0) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
hashdata:
point_indice_data:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
empty_mean:
stream_int:
"""
...
spconv/core_cc/csrc/sparse/all/ops4d/p2v_c.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
class Point2VoxelCommon:
@staticmethod
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]) -> Tuple[List[float], List[int], List[int], List[float]]:
"""
Args:
vsize_xyz:
coors_range_xyz:
"""
...
spconv/core_cc/csrc/sparse/all/ops_cpu1d.pyi
View file @
82fd7a8b
...
...
@@ -9,14 +9,11 @@ class Point2VoxelCPU:
@property
def grid_size(self) -> List[int]: ...
@staticmethod
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]
, num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int
) -> Tuple[List[float], List[int], List[int], List[float]]:
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]) -> Tuple[List[float], List[int], List[int], List[float]]:
"""
Args:
vsize_xyz:
coors_range_xyz:
num_point_features:
max_num_voxels:
max_num_points_per_voxel:
"""
...
def __init__(self, vsize_xyz: List[float], coors_range_xyz: List[float], num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int) -> None:
...
...
@@ -30,7 +27,7 @@ class Point2VoxelCPU:
"""
...
@staticmethod
def point_to_voxel_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor,
mean_per_voxel: Tensor,
vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
def point_to_voxel_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
...
...
@@ -38,7 +35,6 @@ class Point2VoxelCPU:
indices:
num_per_voxel:
densehashdata:
mean_per_voxel:
vsize:
grid_size:
grid_stride:
...
...
@@ -47,7 +43,7 @@ class Point2VoxelCPU:
"""
...
@staticmethod
def point_to_voxel_empty_mean_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor,
mean_per_voxel: Tensor,
vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
def point_to_voxel_empty_mean_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
...
...
@@ -55,7 +51,6 @@ class Point2VoxelCPU:
indices:
num_per_voxel:
densehashdata:
mean_per_voxel:
vsize:
grid_size:
grid_stride:
...
...
spconv/core_cc/csrc/sparse/all/ops_cpu1d/__init__.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
from cumm.tensorview import Tensor
class Point2VoxelCPU:
densehashdata: Tensor
voxels: Tensor
indices: Tensor
num_per_voxel: Tensor
@property
def grid_size(self) -> List[int]: ...
@staticmethod
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]) -> Tuple[List[float], List[int], List[int], List[float]]:
"""
Args:
vsize_xyz:
coors_range_xyz:
"""
...
def __init__(self, vsize_xyz: List[float], coors_range_xyz: List[float], num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int) -> None:
"""
Args:
vsize_xyz:
coors_range_xyz:
num_point_features:
max_num_voxels:
max_num_points_per_voxel:
"""
...
@staticmethod
def point_to_voxel_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
densehashdata:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
"""
...
@staticmethod
def point_to_voxel_empty_mean_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
densehashdata:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
"""
...
def point_to_voxel(self, points: Tensor, clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
"""
...
def point_to_voxel_empty_mean(self, points: Tensor, clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
"""
...
spconv/core_cc/csrc/sparse/all/ops_cpu1d/p2v_c.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
class Point2VoxelCommon:
@staticmethod
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]) -> Tuple[List[float], List[int], List[int], List[float]]:
"""
Args:
vsize_xyz:
coors_range_xyz:
"""
...
spconv/core_cc/csrc/sparse/all/ops_cpu2d.pyi
View file @
82fd7a8b
...
...
@@ -9,14 +9,11 @@ class Point2VoxelCPU:
@property
def grid_size(self) -> List[int]: ...
@staticmethod
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]
, num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int
) -> Tuple[List[float], List[int], List[int], List[float]]:
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]) -> Tuple[List[float], List[int], List[int], List[float]]:
"""
Args:
vsize_xyz:
coors_range_xyz:
num_point_features:
max_num_voxels:
max_num_points_per_voxel:
"""
...
def __init__(self, vsize_xyz: List[float], coors_range_xyz: List[float], num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int) -> None:
...
...
@@ -30,7 +27,7 @@ class Point2VoxelCPU:
"""
...
@staticmethod
def point_to_voxel_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor,
mean_per_voxel: Tensor,
vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
def point_to_voxel_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
...
...
@@ -38,7 +35,6 @@ class Point2VoxelCPU:
indices:
num_per_voxel:
densehashdata:
mean_per_voxel:
vsize:
grid_size:
grid_stride:
...
...
@@ -47,7 +43,7 @@ class Point2VoxelCPU:
"""
...
@staticmethod
def point_to_voxel_empty_mean_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor,
mean_per_voxel: Tensor,
vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
def point_to_voxel_empty_mean_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
...
...
@@ -55,7 +51,6 @@ class Point2VoxelCPU:
indices:
num_per_voxel:
densehashdata:
mean_per_voxel:
vsize:
grid_size:
grid_stride:
...
...
spconv/core_cc/csrc/sparse/all/ops_cpu2d/__init__.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
from cumm.tensorview import Tensor
class Point2VoxelCPU:
densehashdata: Tensor
voxels: Tensor
indices: Tensor
num_per_voxel: Tensor
@property
def grid_size(self) -> List[int]: ...
@staticmethod
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]) -> Tuple[List[float], List[int], List[int], List[float]]:
"""
Args:
vsize_xyz:
coors_range_xyz:
"""
...
def __init__(self, vsize_xyz: List[float], coors_range_xyz: List[float], num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int) -> None:
"""
Args:
vsize_xyz:
coors_range_xyz:
num_point_features:
max_num_voxels:
max_num_points_per_voxel:
"""
...
@staticmethod
def point_to_voxel_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
densehashdata:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
"""
...
@staticmethod
def point_to_voxel_empty_mean_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
voxels:
indices:
num_per_voxel:
densehashdata:
vsize:
grid_size:
grid_stride:
coors_range:
clear_voxels:
"""
...
def point_to_voxel(self, points: Tensor, clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
"""
...
def point_to_voxel_empty_mean(self, points: Tensor, clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
clear_voxels:
"""
...
spconv/core_cc/csrc/sparse/all/ops_cpu2d/p2v_c.pyi
deleted
100644 → 0
View file @
f31eee3a
from typing import overload, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union
from pccm.stubs import EnumValue, EnumClassValue
class Point2VoxelCommon:
@staticmethod
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]) -> Tuple[List[float], List[int], List[int], List[float]]:
"""
Args:
vsize_xyz:
coors_range_xyz:
"""
...
spconv/core_cc/csrc/sparse/all/ops_cpu3d.pyi
View file @
82fd7a8b
...
...
@@ -9,14 +9,11 @@ class Point2VoxelCPU:
@property
def grid_size(self) -> List[int]: ...
@staticmethod
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]
, num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int
) -> Tuple[List[float], List[int], List[int], List[float]]:
def calc_meta_data(vsize_xyz: List[float], coors_range_xyz: List[float]) -> Tuple[List[float], List[int], List[int], List[float]]:
"""
Args:
vsize_xyz:
coors_range_xyz:
num_point_features:
max_num_voxels:
max_num_points_per_voxel:
"""
...
def __init__(self, vsize_xyz: List[float], coors_range_xyz: List[float], num_point_features: int, max_num_voxels: int, max_num_points_per_voxel: int) -> None:
...
...
@@ -30,7 +27,7 @@ class Point2VoxelCPU:
"""
...
@staticmethod
def point_to_voxel_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor,
mean_per_voxel: Tensor,
vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
def point_to_voxel_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
...
...
@@ -38,7 +35,6 @@ class Point2VoxelCPU:
indices:
num_per_voxel:
densehashdata:
mean_per_voxel:
vsize:
grid_size:
grid_stride:
...
...
@@ -47,7 +43,7 @@ class Point2VoxelCPU:
"""
...
@staticmethod
def point_to_voxel_empty_mean_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor,
mean_per_voxel: Tensor,
vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
def point_to_voxel_empty_mean_static(points: Tensor, voxels: Tensor, indices: Tensor, num_per_voxel: Tensor, densehashdata: Tensor, vsize: List[float], grid_size: List[int], grid_stride: List[int], coors_range: List[float], clear_voxels: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
"""
Args:
points:
...
...
@@ -55,7 +51,6 @@ class Point2VoxelCPU:
indices:
num_per_voxel:
densehashdata:
mean_per_voxel:
vsize:
grid_size:
grid_stride:
...
...
Prev
1
2
3
4
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment