|
| FixedRadiusSearchOpKernelCPU (OpKernelConstruction *construction) |
|
void | Kernel (tensorflow::OpKernelContext *context, const tensorflow::Tensor &points, const tensorflow::Tensor &queries, const tensorflow::Tensor &radius, const tensorflow::Tensor &points_row_splits, const tensorflow::Tensor &queries_row_splits, const tensorflow::Tensor &hash_table_splits, const tensorflow::Tensor &hash_table_index, const tensorflow::Tensor &hash_table_cell_splits, tensorflow::Tensor &query_neighbors_row_splits) |
|
◆ FixedRadiusSearchOpKernelCPU()
◆ Kernel()
template<class T >
void FixedRadiusSearchOpKernelCPU< T >::Kernel |
( |
tensorflow::OpKernelContext * |
context, |
|
|
const tensorflow::Tensor & |
points, |
|
|
const tensorflow::Tensor & |
queries, |
|
|
const tensorflow::Tensor & |
radius, |
|
|
const tensorflow::Tensor & |
points_row_splits, |
|
|
const tensorflow::Tensor & |
queries_row_splits, |
|
|
const tensorflow::Tensor & |
hash_table_splits, |
|
|
const tensorflow::Tensor & |
hash_table_index, |
|
|
const tensorflow::Tensor & |
hash_table_cell_splits, |
|
|
tensorflow::Tensor & |
query_neighbors_row_splits |
|
) |
| |
|
inline |
The documentation for this class was generated from the following file: