#include <TrilinearDevoxelizeKernel.h>
◆ TrilinearDevoxelizeGradOpKernel()
TrilinearDevoxelizeGradOpKernel::TrilinearDevoxelizeGradOpKernel |
( |
tensorflow::OpKernelConstruction * |
context | ) |
|
|
inlineexplicit |
◆ Compute()
void TrilinearDevoxelizeGradOpKernel::Compute |
( |
tensorflow::OpKernelContext * |
context | ) |
|
|
inlineoverride |
◆ Kernel()
virtual void TrilinearDevoxelizeGradOpKernel::Kernel |
( |
tensorflow::OpKernelContext * |
context, |
|
|
int |
b, |
|
|
int |
c, |
|
|
int |
n, |
|
|
int |
r3, |
|
|
const int * |
inds, |
|
|
const float * |
wgts, |
|
|
const float * |
grad_y, |
|
|
float * |
grad_x |
|
) |
| |
|
pure virtual |
int TrilinearDevoxelizeGradOpKernel::r |
|
protected |
The documentation for this class was generated from the following file: