std::tuple<Tensor, Tensor> fractional_max_pool2d_cpu(
const at::Tensor& input,
IntArrayRef pool_size,
IntArrayRef output_size,
const at::Tensor& randomSamples)
{
Tensor output = at::empty({0}, input.options());
Tensor indices = at::empty({0}, input.options().dtype(kLong));
fractional_max_pool2d_out_cpu_template(
input,
output,
output_size,
pool_size,
indices,
randomSamples);
return std::tuple<Tensor, Tensor>(output, indices);
}
Tensor fractional_max_pool2d_backward_cpu(
const at::Tensor& gradOutput_,
const at::Tensor& input,
IntArrayRef pool_size,
IntArrayRef output_size,
const at::Tensor& indices)
{
Tensor gradInput = at::empty({0}, input.options());
fractional_max_pool2d_backward_out_cpu_template(
input,
gradOutput_,
gradInput,
output_size,
pool_size,
indices);
return gradInput;
}
