|
static Variable | forward (AutogradContext *ctx, Variable filters, Variable out_importance, Variable inp_features, Variable inp_neighbors_index, Variable inp_neighbors_importance_sum, Variable inp_neighbors_row_splits, Variable neighbors_index, Variable neighbors_kernel_index, Variable neighbors_importance, Variable neighbors_row_splits, const bool normalize, const int64_t max_temp_mem_MB) |
|
static variable_list | backward (AutogradContext *ctx, variable_list grad_output) |
|
◆ backward()
static variable_list SparseConvTransposeFunction::backward |
( |
AutogradContext * |
ctx, |
|
|
variable_list |
grad_output |
|
) |
| |
|
inlinestatic |
◆ forward()
static Variable SparseConvTransposeFunction::forward |
( |
AutogradContext * |
ctx, |
|
|
Variable |
filters, |
|
|
Variable |
out_importance, |
|
|
Variable |
inp_features, |
|
|
Variable |
inp_neighbors_index, |
|
|
Variable |
inp_neighbors_importance_sum, |
|
|
Variable |
inp_neighbors_row_splits, |
|
|
Variable |
neighbors_index, |
|
|
Variable |
neighbors_kernel_index, |
|
|
Variable |
neighbors_importance, |
|
|
Variable |
neighbors_row_splits, |
|
|
const bool |
normalize, |
|
|
const int64_t |
max_temp_mem_MB |
|
) |
| |
|
inlinestatic |
The documentation for this class was generated from the following file: