|
| | ContinuousConvTransposeBackpropFilterOpKernelCPU (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 tensorflow::Tensor &out_features_gradient, 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 &filter_backprop) |
| | ContinuousConvTransposeBackpropFilterOpKernel (tensorflow::OpKernelConstruction *construction) |
| void | Compute (tensorflow::OpKernelContext *context) override |
◆ ContinuousConvTransposeBackpropFilterOpKernelCPU()
template<class TFeat, class TOut, class TReal, class TIndex>
◆ Kernel()
template<class TFeat, class TOut, class TReal, class TIndex>
| void ContinuousConvTransposeBackpropFilterOpKernelCPU< 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 tensorflow::Tensor & | out_features_gradient, |
|
|
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 & | filter_backprop ) |
|
inlinevirtual |
The documentation for this class was generated from the following file: