encode_debug.h 303 Bytes
Newer Older
Nianchen Deng's avatar
sync    
Nianchen Deng committed
1
2
3
4
5
6
7
#pragma once
#include <torch/extension.h>

std::tuple<at::Tensor, at::Tensor, at::Tensor>
multires_hash_encode_debug(const int levels, const int coarse_levels, at::Tensor x,
                           at::Tensor res_list, at::Tensor hash_table,
                           at::Tensor hash_table_offsets);