|
| ContinuousConvTransposeOpKernelCPU (OpKernelConstruction *construction) |
|
void | Kernel (tensorflow::OpKernelContext *context, const tensorflow::Tensor &filter, const tensorflow::Tensor &out_positions, const tensorflow::Tensor &out_importance, const tensorflow::Tensor &extents, const tensorflow::Tensor &offset, const tensorflow::Tensor &inp_positions, const tensorflow::Tensor &inp_features, const tensorflow::Tensor &inp_neighbors_importance_sum, const tensorflow::Tensor &inp_neighbors_row_splits, const tensorflow::Tensor &neighbors_index, const tensorflow::Tensor &neighbors_importance, const tensorflow::Tensor &neighbors_row_splits, const std::vector< int > &filter_dims, const bool individual_extents, const bool isotropic_extents, const bool point_importances, const bool has_neighbors_importances, tensorflow::Tensor &out_features) |
|
| ContinuousConvTransposeOpKernel (tensorflow::OpKernelConstruction *construction) |
|
void | Compute (tensorflow::OpKernelContext *context) override |
|
◆ ContinuousConvTransposeOpKernelCPU()
template<class TFeat , class TOut , class TReal , class TIndex >
◆ Kernel()
template<class TFeat , class TOut , class TReal , class TIndex >
void ContinuousConvTransposeOpKernelCPU< TFeat, TOut, TReal, TIndex >::Kernel |
( |
tensorflow::OpKernelContext * |
context, |
|
|
const tensorflow::Tensor & |
filter, |
|
|
const tensorflow::Tensor & |
out_positions, |
|
|
const tensorflow::Tensor & |
out_importance, |
|
|
const tensorflow::Tensor & |
extents, |
|
|
const tensorflow::Tensor & |
offset, |
|
|
const tensorflow::Tensor & |
inp_positions, |
|
|
const tensorflow::Tensor & |
inp_features, |
|
|
const tensorflow::Tensor & |
inp_neighbors_importance_sum, |
|
|
const tensorflow::Tensor & |
inp_neighbors_row_splits, |
|
|
const tensorflow::Tensor & |
neighbors_index, |
|
|
const tensorflow::Tensor & |
neighbors_importance, |
|
|
const tensorflow::Tensor & |
neighbors_row_splits, |
|
|
const std::vector< int > & |
filter_dims, |
|
|
const bool |
individual_extents, |
|
|
const bool |
isotropic_extents, |
|
|
const bool |
point_importances, |
|
|
const bool |
has_neighbors_importances, |
|
|
tensorflow::Tensor & |
out_features |
|
) |
| |
|
inlinevirtual |
The documentation for this class was generated from the following file: