10#include "torch/script.h"
14 const torch::Tensor& row_splits,
15 const int64_t out_col_size,
16 const torch::Tensor& default_value);
18#ifdef BUILD_CUDA_MODULE
20torch::Tensor RaggedToDenseCUDA(
const torch::Tensor& values,
21 const torch::Tensor& row_splits,
22 const int64_t out_col_size,
23 const torch::Tensor& default_value);
torch::Tensor RaggedToDenseCPU(const torch::Tensor &values, const torch::Tensor &row_splits, const int64_t out_col_size, const torch::Tensor &default_value)
Definition RaggedToDenseOpKernel.cpp:16