You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
797 B
24 lines
797 B
#pragma once
|
|
|
|
// ${generated_comment}
|
|
|
|
#include <ATen/FunctionalStorageImpl.h>
|
|
#include <ATen/Tensor.h>
|
|
|
|
namespace at {
|
|
namespace functionalization {
|
|
|
|
struct FunctionalInverses {
|
|
|
|
${view_inverse_declarations}
|
|
|
|
// NB: These are not generated! They're manually implemented in the template.
|
|
// TODO: Change codegen to generate these. See the following link:
|
|
// https://github.com/pytorch/pytorch/blob/main/torchgen/model.py#L2583-L2585
|
|
static at::Tensor chunk_inverse(const at::Tensor & base, const at::Tensor & mutated_view, InverseReturnMode inverse_return_mode, int64_t mutated_view_idx, int chunks, int dim);
|
|
static at::Tensor narrow_inverse(const at::Tensor & base, const at::Tensor & mutated_view, InverseReturnMode inverse_return_mode, int dim, c10::SymInt start, c10::SymInt length);
|
|
|
|
};
|
|
}
|
|
}
|