|
| | VoxelPoolingGradOpKernelCPU (OpKernelConstruction *construction) |
| void | Kernel (tensorflow::OpKernelContext *context, tensorflow::Tensor &features_backprop, const tensorflow::Tensor &positions, const tensorflow::Tensor &features, const tensorflow::Tensor &pooled_positions, const tensorflow::Tensor &pooled_features_gradient, const tensorflow::Tensor &voxel_size) |
◆ VoxelPoolingGradOpKernelCPU()
template<class TReal, class TFeat>
◆ Kernel()
template<class TReal, class TFeat>
| void VoxelPoolingGradOpKernelCPU< TReal, TFeat >::Kernel |
( |
tensorflow::OpKernelContext * | context, |
|
|
tensorflow::Tensor & | features_backprop, |
|
|
const tensorflow::Tensor & | positions, |
|
|
const tensorflow::Tensor & | features, |
|
|
const tensorflow::Tensor & | pooled_positions, |
|
|
const tensorflow::Tensor & | pooled_features_gradient, |
|
|
const tensorflow::Tensor & | voxel_size ) |
|
inline |
The documentation for this class was generated from the following file: