"...unconditional_image_generation/test_unconditional.py" did not exist on "172acc98b99df702da8c2233f2ec9b57d829ca90"
basis_cpu.cpp 588 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#include "basis_cpu.h"

#include "utils.h"

std::tuple<torch::Tensor, torch::Tensor>
spline_basis_fw_cpu(torch::Tensor pseudo, torch::Tensor kernel_size,
                    torch::Tensor is_open_spline, int64_t degree) {
  return std::make_tuple(pseudo, kernel_size);
}

torch::Tensor spline_basis_bw_cpu(torch::Tensor grad_basis,
                                  torch::Tensor pseudo,
                                  torch::Tensor kernel_size,
                                  torch::Tensor is_open_spline,
                                  int64_t degree) {
  return grad_basis;
}