|
| SparseConvOpKernelCPU (OpKernelConstruction *construction) |
|
void | Kernel (tensorflow::OpKernelContext *context, const tensorflow::Tensor &filters, const tensorflow::Tensor &inp_features, const tensorflow::Tensor &inp_importance, const tensorflow::Tensor &neighbors_index, const tensorflow::Tensor &neighbors_kernel_index, const tensorflow::Tensor &neighbors_importance, const tensorflow::Tensor &neighbors_row_splits, const std::vector< int > &filter_dims, const bool point_importances, const bool has_neighbors_importances, tensorflow::Tensor &out_features) |
|
| SparseConvOpKernel (tensorflow::OpKernelConstruction *construction) |
|
void | Compute (tensorflow::OpKernelContext *context) override |
|
◆ SparseConvOpKernelCPU()
template<class TFeat , class TOut , class TIndex , class TKernelIndex >
◆ Kernel()
template<class TFeat , class TOut , class TIndex , class TKernelIndex >
void SparseConvOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex >::Kernel |
( |
tensorflow::OpKernelContext * |
context, |
|
|
const tensorflow::Tensor & |
filters, |
|
|
const tensorflow::Tensor & |
inp_features, |
|
|
const tensorflow::Tensor & |
inp_importance, |
|
|
const tensorflow::Tensor & |
neighbors_index, |
|
|
const tensorflow::Tensor & |
neighbors_kernel_index, |
|
|
const tensorflow::Tensor & |
neighbors_importance, |
|
|
const tensorflow::Tensor & |
neighbors_row_splits, |
|
|
const std::vector< int > & |
filter_dims, |
|
|
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: