113 template <
class TReal,
class TIndex>
120 const TReal*
const __restrict__ out_positions,
122 const TReal*
const __restrict__ inp_positions,
123 const TReal*
const __restrict__ inp_features,
124 const TReal*
const __restrict__ inp_importance,
125 size_t neighbors_index_size,
126 const TIndex*
const __restrict__ neighbors_index,
127 const TReal*
const __restrict__ neighbors_importance,
128 const int64_t*
const __restrict__ neighbors_row_splits,
129 const TReal*
const __restrict__ extents,
130 const TReal*
const __restrict__ offsets,
131 const std::vector<int>& filter_dims,
135 bool individual_extent,
136 bool isotropic_extent,
139 template <
class TReal,
class TIndex>
141 const cudaStream_t& stream,
147 const TReal*
const __restrict__ out_positions,
149 const TReal*
const __restrict__ inp_positions,
150 const TReal*
const __restrict__ inp_features,
151 const TReal*
const __restrict__ inp_neighbors_importance_sum,
152 const int64_t*
const __restrict__ inp_neighbors_prefix_sum,
153 size_t neighbors_index_size,
154 const TIndex*
const __restrict__ neighbors_index,
155 const TReal*
const __restrict__ neighbors_importance,
156 const int64_t*
const __restrict__ neighbors_row_splits,
157 const TReal*
const __restrict__ extents,
158 const TReal*
const __restrict__ offsets,
159 const std::vector<int>& filter_dims,
163 bool individual_extent,
164 bool isotropic_extent,
178 T* __restrict__ col_major_matrix,
179 const T*
const __restrict__ vector);
194 T* __restrict__ out_ptr,
195 const T*
const __restrict__ col_major_matrix,
196 const T*
const __restrict__ vector);
InterpolationMode
Definition: ContinuousConvTypes.h:37
void FillColumn(const cudaStream_t &stream, TReal *columns, int in_channels, TIndex begin_idx, TIndex end_idx, TIndex num_out, const TReal *const __restrict__ out_positions, TIndex num_inp, const TReal *const __restrict__ inp_positions, const TReal *const __restrict__ inp_features, const TReal *const __restrict__ inp_importance, size_t neighbors_index_size, const TIndex *const __restrict__ neighbors_index, const TReal *const __restrict__ neighbors_importance, const int64_t *const __restrict__ neighbors_row_splits, const TReal *const __restrict__ extents, const TReal *const __restrict__ offsets, const std::vector< int > &filter_dims, InterpolationMode interpolation, CoordinateMapping coordinate_mapping, bool align_corners, bool individual_extent, bool isotropic_extent, bool normalize)
void FillColumnTranspose(const cudaStream_t &stream, TReal *columns, int in_channels, TIndex begin_idx, TIndex end_idx, TIndex num_out, const TReal *const __restrict__ out_positions, TIndex num_inp, const TReal *const __restrict__ inp_positions, const TReal *const __restrict__ inp_features, const TReal *const __restrict__ inp_neighbors_importance_sum, const int64_t *const __restrict__ inp_neighbors_prefix_sum, size_t neighbors_index_size, const TIndex *const __restrict__ neighbors_index, const TReal *const __restrict__ neighbors_importance, const int64_t *const __restrict__ neighbors_row_splits, const TReal *const __restrict__ extents, const TReal *const __restrict__ offsets, const std::vector< int > &filter_dims, InterpolationMode interpolation, CoordinateMapping coordinate_mapping, bool align_corners, bool individual_extent, bool isotropic_extent, bool normalize)
void MultiplyAndCopyColumns(const cudaStream_t &stream, size_t rows, size_t cols, T *__restrict__ out_ptr, const T *const __restrict__ col_major_matrix, const T *const __restrict__ vector)
CoordinateMapping
Definition: ContinuousConvTypes.h:45
Definition: PinholeCameraIntrinsic.cpp:35
void MultiplyColumns(const cudaStream_t &stream, size_t rows, size_t cols, T *__restrict__ col_major_matrix, const T *const __restrict__ vector)