ContinuousConvTranspose.h File Reference

(313315d (Wed Dec 23 23:39:47 2020 -0800))

#include <tbb/parallel_for.h>
#include "open3d/ml/impl/continuous_conv/CoordinateTransformation.h"

Go to the source code of this file.

Namespaces
	open3d
	open3d::ml
	open3d::ml::impl

Macros
#define	FN_PARAMETERS
#define	CALL_TEMPLATE(INTERPOLATION, MAPPING, ALIGN_CORNERS, INDIVIDUAL_EXTENT, ISOTROPIC_EXTENT, NORMALIZE)
#define	CALL_TEMPLATE2(INTERPOLATION, MAPPING)
#define	CALL_TEMPLATE3(INTERPOLATION)
#define	CALL_TEMPLATE4

Functions
template<class TReal , class TIndex , InterpolationMode INTERPOLATION, CoordinateMapping MAPPING, bool ALIGN_CORNERS, bool INDIVIDUAL_EXTENT, bool ISOTROPIC_EXTENT, bool NORMALIZE>
void	open3d::ml::impl::_CConvTransposeComputeFeaturesCPU (TReal *out_features, const std::vector< int > &filter_dims, const TReal *filter, size_t num_out, const TReal *out_positions, const TReal *out_importance, size_t num_inp, const TReal *inp_positions, const TReal *inp_features, const TReal *inp_neighbors_importance_sum, const int64_t *inp_neighbors_row_splits, size_t neighbors_index_size, const TIndex *neighbor_index, const TReal *neighbor_importance, const int64_t *neighbors_row_splits, const TReal *extents, const TReal *offsets)
template<class TReal , class TIndex >
void	open3d::ml::impl::CConvTransposeComputeFeaturesCPU (TReal *out_features, const std::vector< int > &filter_dims, const TReal *filter, size_t num_out, const TReal *out_positions, const TReal *out_importance, size_t num_inp, const TReal *inp_positions, const TReal *inp_features, const TReal *inp_neighbors_importance_sum, const int64_t *inp_neighbors_row_splits, size_t neighbors_index_size, const TIndex *neighbor_index, const TReal *neighbor_importance, const int64_t *neighbors_row_splits, const TReal *extents, const TReal *offsets, InterpolationMode interpolation, CoordinateMapping coordinate_mapping, bool align_corners, bool individual_extent, bool isotropic_extent, bool normalize)

Macro Definition Documentation

CALL_TEMPLATE

#define CALL_TEMPLATE	(		INTERPOLATION,
			MAPPING,
			ALIGN_CORNERS,
			INDIVIDUAL_EXTENT,
			ISOTROPIC_EXTENT,
			NORMALIZE
	)

Value:

if (INTERPOLATION == interpolation && MAPPING == coordinate_mapping && \
        ALIGN_CORNERS == align_corners &&                                  \
        INDIVIDUAL_EXTENT == individual_extent &&                          \
        ISOTROPIC_EXTENT == isotropic_extent && NORMALIZE == normalize)    \
        _CConvTransposeComputeFeaturesCPU<                                 \
                TReal, TIndex, INTERPOLATION, MAPPING, ALIGN_CORNERS,      \
                INDIVIDUAL_EXTENT, ISOTROPIC_EXTENT, NORMALIZE>(           \
                FN_PARAMETERS);

CALL_TEMPLATE2

#define CALL_TEMPLATE2	(		INTERPOLATION,
			MAPPING
	)

Value:

CALL_TEMPLATE(INTERPOLATION, MAPPING, true, true, true, true)    \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, true, true, true, false)   \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, true, true, false, true)   \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, true, true, false, false)  \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, true, false, true, true)   \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, true, false, true, false)  \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, true, false, false, true)  \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, true, false, false, false) \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, false, true, true, true)   \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, false, true, true, false)  \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, false, true, false, true)  \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, false, true, false, false) \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, false, false, true, true)  \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, false, false, true, false) \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, false, false, false, true) \
    CALL_TEMPLATE(INTERPOLATION, MAPPING, false, false, false, false)

CALL_TEMPLATE3

#define CALL_TEMPLATE3

(

INTERPOLATION

)

Value:

CALL_TEMPLATE2(INTERPOLATION, CoordinateMapping::BALL_TO_CUBE_RADIAL) \
    CALL_TEMPLATE2(INTERPOLATION,                                         \
                   CoordinateMapping::BALL_TO_CUBE_VOLUME_PRESERVING)     \
    CALL_TEMPLATE2(INTERPOLATION, CoordinateMapping::IDENTITY)

CALL_TEMPLATE4

#define CALL_TEMPLATE4

Value:

CALL_TEMPLATE3(InterpolationMode::LINEAR)        \
    CALL_TEMPLATE3(InterpolationMode::LINEAR_BORDER) \
    CALL_TEMPLATE3(InterpolationMode::NEAREST_NEIGHBOR)

FN_PARAMETERS

#define FN_PARAMETERS

Value:

out_features, filter_dims, filter, num_out, out_positions, out_importance, \
            num_inp, inp_positions, inp_features,                              \
            inp_neighbors_importance_sum, inp_neighbors_row_splits,            \
            neighbors_index_size, neighbor_index, neighbor_importance,         \
            neighbors_row_splits, extents, offsets