|
template<typename T , typename = enable_if_symint<T>> |
c10::SymIntArrayRef | sizes (const TensorBase &t) |
|
template<typename T , typename = enable_if_int<T>> |
IntArrayRef | sizes (const TensorBase &t) |
|
template<typename T , typename = enable_if_symint<T>> |
c10::SymInt | size (const TensorBase &t, int64_t dim) |
|
template<typename T , typename = enable_if_int<T>> |
int64_t | size (const TensorBase &t, int64_t dim) |
|
template<typename T , typename = enable_if_symint<T>> |
c10::SymIntArrayRef | strides (const TensorBase &t) |
|
template<typename T , typename = enable_if_int<T>> |
IntArrayRef | strides (const TensorBase &t) |
|
template<typename T , typename = enable_if_symint<T>> |
c10::SymInt | numel (const TensorBase &t) |
|
template<typename T , typename = enable_if_int<T>> |
int64_t | numel (const TensorBase &t) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _cudnn_rnn_flatten_weight (at::TensorList weight_arr, int64_t weight_stride0, int64_t input_size, int64_t mode, int64_t hidden_size, int64_t proj_size, int64_t num_layers, bool batch_first, bool bidirectional) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _cudnn_rnn_flatten_weight (at::TensorList weight_arr, int64_t weight_stride0, c10::SymInt input_size, int64_t mode, c10::SymInt hidden_size, c10::SymInt proj_size, int64_t num_layers, bool batch_first, bool bidirectional) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor, at::Tensor, at::Tensor > | _cudnn_rnn (const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const c10::optional< at::Tensor > &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, int64_t mode, int64_t hidden_size, int64_t proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, at::IntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor, at::Tensor, at::Tensor > | _cudnn_rnn (const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const c10::optional< at::Tensor > &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, int64_t mode, c10::SymInt hidden_size, c10::SymInt proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, c10::SymIntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor,::std::vector< at::Tensor > > | _cudnn_rnn_backward (const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const at::Tensor &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, const at::Tensor &output, const c10::optional< at::Tensor > &grad_output, const c10::optional< at::Tensor > &grad_hy, const c10::optional< at::Tensor > &grad_cy, int64_t mode, int64_t hidden_size, int64_t proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, at::IntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state, const at::Tensor &reserve, ::std::array< bool, 4 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor,::std::vector< at::Tensor > > | _cudnn_rnn_backward (const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const at::Tensor &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, const at::Tensor &output, const c10::optional< at::Tensor > &grad_output, const c10::optional< at::Tensor > &grad_hy, const c10::optional< at::Tensor > &grad_cy, int64_t mode, c10::SymInt hidden_size, c10::SymInt proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, c10::SymIntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state, const at::Tensor &reserve, ::std::array< bool, 4 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | as_strided (const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride, c10::optional< int64_t > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | as_strided (const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, c10::optional< c10::SymInt > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
const at::Tensor & | as_strided_ (const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride, c10::optional< int64_t > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
const at::Tensor & | as_strided_ (const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, c10::optional< c10::SymInt > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | broadcast_to (const at::Tensor &self, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | broadcast_to (const at::Tensor &self, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::vector< at::Tensor > | tensor_split (const at::Tensor &self, int64_t sections, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::vector< at::Tensor > | tensor_split (const at::Tensor &self, c10::SymInt sections, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::vector< at::Tensor > | tensor_split (const at::Tensor &self, at::IntArrayRef indices, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::vector< at::Tensor > | tensor_split (const at::Tensor &self, c10::SymIntArrayRef indices, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | constant_pad_nd (const at::Tensor &self, at::IntArrayRef pad, const at::Scalar &value=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | constant_pad_nd (const at::Tensor &self, c10::SymIntArrayRef pad, const at::Scalar &value=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | convolution (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool transposed, at::IntArrayRef output_padding, int64_t groups) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | convolution (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, bool transposed, c10::SymIntArrayRef output_padding, int64_t groups) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | convolution_backward (const at::Tensor &grad_output, const at::Tensor &input, const at::Tensor &weight, at::OptionalIntArrayRef bias_sizes, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool transposed, at::IntArrayRef output_padding, int64_t groups, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | convolution_backward (const at::Tensor &grad_output, const at::Tensor &input, const at::Tensor &weight, at::OptionalSymIntArrayRef bias_sizes, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, bool transposed, c10::SymIntArrayRef output_padding, int64_t groups, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _convolution (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool transposed, at::IntArrayRef output_padding, int64_t groups, bool benchmark, bool deterministic, bool cudnn_enabled, bool allow_tf32) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _convolution (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, bool transposed, c10::SymIntArrayRef output_padding, int64_t groups, bool benchmark, bool deterministic, bool cudnn_enabled, bool allow_tf32) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | _convolution_double_backward (const c10::optional< at::Tensor > &ggI, const c10::optional< at::Tensor > &ggW, const c10::optional< at::Tensor > &ggb, const at::Tensor &gO, const at::Tensor &weight, const at::Tensor &self, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool transposed, at::IntArrayRef output_padding, int64_t groups, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | _convolution_double_backward (const c10::optional< at::Tensor > &ggI, const c10::optional< at::Tensor > &ggW, const c10::optional< at::Tensor > &ggb, const at::Tensor &gO, const at::Tensor &weight, const at::Tensor &self, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, bool transposed, c10::SymIntArrayRef output_padding, int64_t groups, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | diagonal_backward (const at::Tensor &grad_output, at::IntArrayRef input_sizes, int64_t offset, int64_t dim1, int64_t dim2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | diagonal_backward (const at::Tensor &grad_output, c10::SymIntArrayRef input_sizes, int64_t offset, int64_t dim1, int64_t dim2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | embedding (const at::Tensor &weight, const at::Tensor &indices, int64_t padding_idx=-1, bool scale_grad_by_freq=false, bool sparse=false) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | embedding (const at::Tensor &weight, const at::Tensor &indices, c10::SymInt padding_idx=-1, bool scale_grad_by_freq=false, bool sparse=false) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | embedding_backward (const at::Tensor &grad, const at::Tensor &indices, int64_t num_weights, int64_t padding_idx, bool scale_grad_by_freq, bool sparse) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | embedding_backward (const at::Tensor &grad, const at::Tensor &indices, c10::SymInt num_weights, c10::SymInt padding_idx, bool scale_grad_by_freq, bool sparse) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | embedding_dense_backward (const at::Tensor &grad_output, const at::Tensor &indices, int64_t num_weights, int64_t padding_idx, bool scale_grad_by_freq) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | embedding_dense_backward (const at::Tensor &grad_output, const at::Tensor &indices, c10::SymInt num_weights, c10::SymInt padding_idx, bool scale_grad_by_freq) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _embedding_bag_backward (const at::Tensor &grad, const at::Tensor &indices, const at::Tensor &offsets, const at::Tensor &offset2bag, const at::Tensor &bag_size, const at::Tensor &maximum_indices, int64_t num_weights, bool scale_grad_by_freq, int64_t mode, bool sparse, const c10::optional< at::Tensor > &per_sample_weights, int64_t padding_idx=-1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _embedding_bag_backward (const at::Tensor &grad, const at::Tensor &indices, const at::Tensor &offsets, const at::Tensor &offset2bag, const at::Tensor &bag_size, const at::Tensor &maximum_indices, c10::SymInt num_weights, bool scale_grad_by_freq, int64_t mode, bool sparse, const c10::optional< at::Tensor > &per_sample_weights, int64_t padding_idx=-1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _embedding_bag_sparse_backward (const at::Tensor &grad, const at::Tensor &indices, const at::Tensor &offsets, const at::Tensor &offset2bag, const at::Tensor &bag_size, int64_t num_weights, bool scale_grad_by_freq, int64_t mode, const c10::optional< at::Tensor > &per_sample_weights, int64_t padding_idx=-1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _embedding_bag_sparse_backward (const at::Tensor &grad, const at::Tensor &indices, const at::Tensor &offsets, const at::Tensor &offset2bag, const at::Tensor &bag_size, c10::SymInt num_weights, bool scale_grad_by_freq, int64_t mode, const c10::optional< at::Tensor > &per_sample_weights, int64_t padding_idx=-1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _embedding_bag_dense_backward (const at::Tensor &grad, const at::Tensor &indices, const at::Tensor &offset2bag, const at::Tensor &bag_size, const at::Tensor &maximum_indices, int64_t num_weights, bool scale_grad_by_freq, int64_t mode, const c10::optional< at::Tensor > &per_sample_weights, int64_t padding_idx=-1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _embedding_bag_dense_backward (const at::Tensor &grad, const at::Tensor &indices, const at::Tensor &offset2bag, const at::Tensor &bag_size, const at::Tensor &maximum_indices, c10::SymInt num_weights, bool scale_grad_by_freq, int64_t mode, const c10::optional< at::Tensor > &per_sample_weights, int64_t padding_idx=-1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | empty (at::IntArrayRef size, at::TensorOptions options={}, c10::optional< at::MemoryFormat > memory_format=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | empty (at::IntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory, c10::optional< at::MemoryFormat > memory_format) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | empty (c10::SymIntArrayRef size, at::TensorOptions options={}, c10::optional< at::MemoryFormat > memory_format=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | empty (c10::SymIntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory, c10::optional< at::MemoryFormat > memory_format) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | new_empty (const at::Tensor &self, at::IntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | new_empty (const at::Tensor &self, at::IntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | new_empty (const at::Tensor &self, c10::SymIntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | new_empty (const at::Tensor &self, c10::SymIntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | new_empty_strided (const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | new_empty_strided (const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | new_empty_strided (const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | new_empty_strided (const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | new_full (const at::Tensor &self, at::IntArrayRef size, const at::Scalar &fill_value, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | new_full (const at::Tensor &self, at::IntArrayRef size, const at::Scalar &fill_value, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | new_full (const at::Tensor &self, c10::SymIntArrayRef size, const at::Scalar &fill_value, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | new_full (const at::Tensor &self, c10::SymIntArrayRef size, const at::Scalar &fill_value, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | new_zeros (const at::Tensor &self, at::IntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | new_zeros (const at::Tensor &self, at::IntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | new_zeros (const at::Tensor &self, c10::SymIntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | new_zeros (const at::Tensor &self, c10::SymIntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | new_ones (const at::Tensor &self, at::IntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | new_ones (const at::Tensor &self, at::IntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | new_ones (const at::Tensor &self, c10::SymIntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | new_ones (const at::Tensor &self, c10::SymIntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
const at::Tensor & | resize_ (const at::Tensor &self, at::IntArrayRef size, c10::optional< at::MemoryFormat > memory_format=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
const at::Tensor & | resize_ (const at::Tensor &self, c10::SymIntArrayRef size, c10::optional< at::MemoryFormat > memory_format=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | empty_out (at::Tensor &out, at::IntArrayRef size, c10::optional< at::MemoryFormat > memory_format=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | empty_outf (at::IntArrayRef size, c10::optional< at::MemoryFormat > memory_format, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | empty_out (at::Tensor &out, c10::SymIntArrayRef size, c10::optional< at::MemoryFormat > memory_format=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | empty_outf (c10::SymIntArrayRef size, c10::optional< at::MemoryFormat > memory_format, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | empty_strided (at::IntArrayRef size, at::IntArrayRef stride, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | empty_strided (at::IntArrayRef size, at::IntArrayRef stride, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | empty_strided (c10::SymIntArrayRef size, c10::SymIntArrayRef stride, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | empty_strided (c10::SymIntArrayRef size, c10::SymIntArrayRef stride, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | expand (const at::Tensor &self, at::IntArrayRef size, bool implicit=false) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | expand (const at::Tensor &self, c10::SymIntArrayRef size, bool implicit=false) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | full (at::IntArrayRef size, const at::Scalar &fill_value, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | full (at::IntArrayRef size, const at::Scalar &fill_value, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | full (c10::SymIntArrayRef size, const at::Scalar &fill_value, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | full (c10::SymIntArrayRef size, const at::Scalar &fill_value, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | full_out (at::Tensor &out, at::IntArrayRef size, const at::Scalar &fill_value) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | full_outf (at::IntArrayRef size, const at::Scalar &fill_value, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | full_out (at::Tensor &out, c10::SymIntArrayRef size, const at::Scalar &fill_value) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | full_outf (c10::SymIntArrayRef size, const at::Scalar &fill_value, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | native_group_norm (const at::Tensor &input, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, int64_t N, int64_t C, int64_t HxW, int64_t group, double eps) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | native_group_norm (const at::Tensor &input, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, c10::SymInt N, c10::SymInt C, c10::SymInt HxW, int64_t group, double eps) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | native_group_norm_backward (const at::Tensor &grad_out, const at::Tensor &input, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, int64_t N, int64_t C, int64_t HxW, int64_t group, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | native_group_norm_backward (const at::Tensor &grad_out, const at::Tensor &input, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, c10::SymInt N, c10::SymInt C, c10::SymInt HxW, int64_t group, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _fft_c2c (const at::Tensor &self, at::IntArrayRef dim, int64_t normalization, bool forward) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _fft_c2c (const at::Tensor &self, c10::SymIntArrayRef dim, int64_t normalization, bool forward) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _fft_c2c_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef dim, int64_t normalization, bool forward) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _fft_c2c_outf (const at::Tensor &self, at::IntArrayRef dim, int64_t normalization, bool forward, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _fft_c2c_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef dim, int64_t normalization, bool forward) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _fft_c2c_outf (const at::Tensor &self, c10::SymIntArrayRef dim, int64_t normalization, bool forward, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | layer_norm (const at::Tensor &input, at::IntArrayRef normalized_shape, const c10::optional< at::Tensor > &weight={}, const c10::optional< at::Tensor > &bias={}, double eps=1e-05, bool cudnn_enable=true) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | layer_norm (const at::Tensor &input, c10::SymIntArrayRef normalized_shape, const c10::optional< at::Tensor > &weight={}, const c10::optional< at::Tensor > &bias={}, double eps=1e-05, bool cudnn_enable=true) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | native_layer_norm (const at::Tensor &input, at::IntArrayRef normalized_shape, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, double eps) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | native_layer_norm (const at::Tensor &input, c10::SymIntArrayRef normalized_shape, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, double eps) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | native_layer_norm_backward (const at::Tensor &grad_out, const at::Tensor &input, at::IntArrayRef normalized_shape, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor, at::Tensor, at::Tensor > | native_layer_norm_backward (const at::Tensor &grad_out, const at::Tensor &input, c10::SymIntArrayRef normalized_shape, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | value_selecting_reduction_backward (const at::Tensor &grad, int64_t dim, const at::Tensor &indices, at::IntArrayRef sizes, bool keepdim) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | value_selecting_reduction_backward (const at::Tensor &grad, int64_t dim, const at::Tensor &indices, c10::SymIntArrayRef sizes, bool keepdim) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | mkldnn_convolution (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | mkldnn_convolution (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | miopen_convolution (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | miopen_convolution (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | miopen_convolution_transpose (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef output_padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | miopen_convolution_transpose (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, c10::SymIntArrayRef output_padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | miopen_depthwise_convolution (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | miopen_depthwise_convolution (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | narrow_copy (const at::Tensor &self, int64_t dim, int64_t start, int64_t length) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | narrow_copy (const at::Tensor &self, int64_t dim, c10::SymInt start, c10::SymInt length) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | narrow_copy_out (at::Tensor &out, const at::Tensor &self, int64_t dim, int64_t start, int64_t length) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | narrow_copy_outf (const at::Tensor &self, int64_t dim, int64_t start, int64_t length, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | narrow_copy_out (at::Tensor &out, const at::Tensor &self, int64_t dim, c10::SymInt start, c10::SymInt length) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | narrow_copy_outf (const at::Tensor &self, int64_t dim, c10::SymInt start, c10::SymInt length, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | narrow (const at::Tensor &self, int64_t dim, int64_t start, int64_t length) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | narrow (const at::Tensor &self, int64_t dim, c10::SymInt start, c10::SymInt length) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | narrow (const at::Tensor &self, int64_t dim, const at::Tensor &start, int64_t length) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | narrow (const at::Tensor &self, int64_t dim, const at::Tensor &start, c10::SymInt length) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _nnpack_spatial_convolution (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _nnpack_spatial_convolution (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | ones (at::IntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | ones (at::IntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | ones (c10::SymIntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | ones (c10::SymIntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | ones_out (at::Tensor &out, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | ones_outf (at::IntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | ones_out (at::Tensor &out, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | ones_outf (c10::SymIntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | rand (at::IntArrayRef size, c10::optional< at::DimnameList > names, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | rand (at::IntArrayRef size, c10::optional< at::DimnameList > names, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | rand (c10::SymIntArrayRef size, c10::optional< at::DimnameList > names, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | rand (c10::SymIntArrayRef size, c10::optional< at::DimnameList > names, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | rand (at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | rand (at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | rand (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | rand (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | rand (at::IntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | rand (at::IntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | rand (c10::SymIntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | rand (c10::SymIntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | rand (at::IntArrayRef size, c10::optional< at::Generator > generator, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | rand (at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | rand (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | rand (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | rand_out (at::Tensor &out, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | rand_outf (at::IntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | rand_out (at::Tensor &out, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | rand_outf (c10::SymIntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | rand_out (at::Tensor &out, at::IntArrayRef size, c10::optional< at::Generator > generator) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | rand_outf (at::IntArrayRef size, c10::optional< at::Generator > generator, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | rand_out (at::Tensor &out, c10::SymIntArrayRef size, c10::optional< at::Generator > generator) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | rand_outf (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randint (int64_t high, at::IntArrayRef size, at::TensorOptions options=at::kLong) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randint (int64_t high, at::IntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randint (int64_t high, c10::SymIntArrayRef size, at::TensorOptions options=at::kLong) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randint (int64_t high, c10::SymIntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randint (int64_t high, at::IntArrayRef size, c10::optional< at::Generator > generator, at::TensorOptions options=at::kLong) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randint (int64_t high, at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randint (int64_t high, c10::SymIntArrayRef size, c10::optional< at::Generator > generator, at::TensorOptions options=at::kLong) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randint (int64_t high, c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randint (int64_t low, int64_t high, at::IntArrayRef size, at::TensorOptions options=at::kLong) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randint (int64_t low, int64_t high, at::IntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randint (int64_t low, int64_t high, c10::SymIntArrayRef size, at::TensorOptions options=at::kLong) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randint (int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randint (int64_t low, int64_t high, at::IntArrayRef size, c10::optional< at::Generator > generator, at::TensorOptions options=at::kLong) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randint (int64_t low, int64_t high, at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randint (int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional< at::Generator > generator, at::TensorOptions options=at::kLong) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randint (int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randint_out (at::Tensor &out, int64_t high, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randint_outf (int64_t high, at::IntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randint_out (at::Tensor &out, int64_t high, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randint_outf (int64_t high, c10::SymIntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randint_out (at::Tensor &out, int64_t high, at::IntArrayRef size, c10::optional< at::Generator > generator) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randint_outf (int64_t high, at::IntArrayRef size, c10::optional< at::Generator > generator, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randint_out (at::Tensor &out, int64_t high, c10::SymIntArrayRef size, c10::optional< at::Generator > generator) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randint_outf (int64_t high, c10::SymIntArrayRef size, c10::optional< at::Generator > generator, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randint_out (at::Tensor &out, int64_t low, int64_t high, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randint_outf (int64_t low, int64_t high, at::IntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randint_out (at::Tensor &out, int64_t low, int64_t high, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randint_outf (int64_t low, int64_t high, c10::SymIntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randint_out (at::Tensor &out, int64_t low, int64_t high, at::IntArrayRef size, c10::optional< at::Generator > generator) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randint_outf (int64_t low, int64_t high, at::IntArrayRef size, c10::optional< at::Generator > generator, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randint_out (at::Tensor &out, int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional< at::Generator > generator) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randint_outf (int64_t low, int64_t high, c10::SymIntArrayRef size, c10::optional< at::Generator > generator, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randn (at::IntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randn (at::IntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randn (c10::SymIntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randn (c10::SymIntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randn (at::IntArrayRef size, c10::optional< at::Generator > generator, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randn (at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randn (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randn (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randn (at::IntArrayRef size, c10::optional< at::DimnameList > names, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randn (at::IntArrayRef size, c10::optional< at::DimnameList > names, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randn (c10::SymIntArrayRef size, c10::optional< at::DimnameList > names, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randn (c10::SymIntArrayRef size, c10::optional< at::DimnameList > names, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randn (at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | randn (at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randn (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | randn (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randn_out (at::Tensor &out, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randn_outf (at::IntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randn_out (at::Tensor &out, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randn_outf (c10::SymIntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randn_out (at::Tensor &out, at::IntArrayRef size, c10::optional< at::Generator > generator) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randn_outf (at::IntArrayRef size, c10::optional< at::Generator > generator, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randn_out (at::Tensor &out, c10::SymIntArrayRef size, c10::optional< at::Generator > generator) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randn_outf (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | repeat (const at::Tensor &self, at::IntArrayRef repeats) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | repeat (const at::Tensor &self, c10::SymIntArrayRef repeats) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | repeat_interleave (const at::Tensor &self, int64_t repeats, c10::optional< int64_t > dim=c10::nullopt, c10::optional< int64_t > output_size=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | repeat_interleave (const at::Tensor &self, c10::SymInt repeats, c10::optional< int64_t > dim=c10::nullopt, c10::optional< int64_t > output_size=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | reshape (const at::Tensor &self, at::IntArrayRef shape) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | reshape (const at::Tensor &self, c10::SymIntArrayRef shape) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _reshape_copy (const at::Tensor &self, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _reshape_copy (const at::Tensor &self, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _reshape_alias (const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _reshape_alias (const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | select (const at::Tensor &self, int64_t dim, int64_t index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | select (const at::Tensor &self, int64_t dim, c10::SymInt index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | select_backward (const at::Tensor &grad_output, at::IntArrayRef input_sizes, int64_t dim, int64_t index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | select_backward (const at::Tensor &grad_output, c10::SymIntArrayRef input_sizes, int64_t dim, c10::SymInt index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _nested_select_backward (const at::Tensor &grad_output, const at::Tensor &self, int64_t dim, int64_t index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _nested_select_backward (const at::Tensor &grad_output, const at::Tensor &self, int64_t dim, c10::SymInt index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | slice (const at::Tensor &self, int64_t dim=0, c10::optional< int64_t > start=c10::nullopt, c10::optional< int64_t > end=c10::nullopt, int64_t step=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | slice (const at::Tensor &self, int64_t dim=0, c10::optional< c10::SymInt > start=c10::nullopt, c10::optional< c10::SymInt > end=c10::nullopt, c10::SymInt step=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | slice_backward (const at::Tensor &grad_output, at::IntArrayRef input_sizes, int64_t dim, int64_t start, int64_t end, int64_t step) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | slice_backward (const at::Tensor &grad_output, c10::SymIntArrayRef input_sizes, int64_t dim, c10::SymInt start, c10::SymInt end, c10::SymInt step) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | slice_scatter (const at::Tensor &self, const at::Tensor &src, int64_t dim=0, c10::optional< int64_t > start=c10::nullopt, c10::optional< int64_t > end=c10::nullopt, int64_t step=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | slice_scatter (const at::Tensor &self, const at::Tensor &src, int64_t dim=0, c10::optional< c10::SymInt > start=c10::nullopt, c10::optional< c10::SymInt > end=c10::nullopt, c10::SymInt step=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | select_scatter (const at::Tensor &self, const at::Tensor &src, int64_t dim, int64_t index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | select_scatter (const at::Tensor &self, const at::Tensor &src, int64_t dim, c10::SymInt index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | as_strided_scatter (const at::Tensor &self, const at::Tensor &src, at::IntArrayRef size, at::IntArrayRef stride, c10::optional< int64_t > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | as_strided_scatter (const at::Tensor &self, const at::Tensor &src, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, c10::optional< c10::SymInt > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::vector< at::Tensor > | unsafe_split (const at::Tensor &self, int64_t split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::vector< at::Tensor > | unsafe_split (const at::Tensor &self, c10::SymInt split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::vector< at::Tensor > | split (const at::Tensor &self, int64_t split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::vector< at::Tensor > | split (const at::Tensor &self, c10::SymInt split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::vector< at::Tensor > | split (const at::Tensor &self, at::IntArrayRef split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::vector< at::Tensor > | split (const at::Tensor &self, c10::SymIntArrayRef split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::vector< at::Tensor > | unsafe_split_with_sizes (const at::Tensor &self, at::IntArrayRef split_sizes, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::vector< at::Tensor > | unsafe_split_with_sizes (const at::Tensor &self, c10::SymIntArrayRef split_sizes, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::vector< at::Tensor > | split_with_sizes (const at::Tensor &self, at::IntArrayRef split_sizes, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::vector< at::Tensor > | split_with_sizes (const at::Tensor &self, c10::SymIntArrayRef split_sizes, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _unsafe_view (const at::Tensor &self, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _unsafe_view (const at::Tensor &self, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | zeros (at::IntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | zeros (at::IntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | zeros (c10::SymIntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | zeros (c10::SymIntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | zeros_out (at::Tensor &out, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | zeros_outf (at::IntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | zeros_out (at::Tensor &out, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | zeros_outf (c10::SymIntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _sparse_coo_tensor_unsafe (const at::Tensor &indices, const at::Tensor &values, at::IntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _sparse_coo_tensor_unsafe (const at::Tensor &indices, const at::Tensor &values, at::IntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _sparse_coo_tensor_unsafe (const at::Tensor &indices, const at::Tensor &values, c10::SymIntArrayRef size, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _sparse_coo_tensor_unsafe (const at::Tensor &indices, const at::Tensor &values, c10::SymIntArrayRef size, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _sparse_coo_tensor_with_dims_and_tensors (int64_t sparse_dim, int64_t dense_dim, at::IntArrayRef size, const at::Tensor &indices, const at::Tensor &values, at::TensorOptions options) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _sparse_coo_tensor_with_dims_and_tensors (int64_t sparse_dim, int64_t dense_dim, at::IntArrayRef size, const at::Tensor &indices, const at::Tensor &values, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _sparse_coo_tensor_with_dims_and_tensors (int64_t sparse_dim, int64_t dense_dim, c10::SymIntArrayRef size, const at::Tensor &indices, const at::Tensor &values, at::TensorOptions options) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _sparse_coo_tensor_with_dims_and_tensors (int64_t sparse_dim, int64_t dense_dim, c10::SymIntArrayRef size, const at::Tensor &indices, const at::Tensor &values, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _pack_padded_sequence_backward (const at::Tensor &grad, at::IntArrayRef input_size, const at::Tensor &batch_sizes, bool batch_first) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _pack_padded_sequence_backward (const at::Tensor &grad, c10::SymIntArrayRef input_size, const at::Tensor &batch_sizes, bool batch_first) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | set_ (at::Tensor &self, at::Storage source, int64_t storage_offset, at::IntArrayRef size, at::IntArrayRef stride={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | set_ (at::Tensor &self, at::Storage source, c10::SymInt storage_offset, c10::SymIntArrayRef size, c10::SymIntArrayRef stride={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | set_ (at::Tensor &self, const at::Tensor &source, int64_t storage_offset, at::IntArrayRef size, at::IntArrayRef stride={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | set_ (at::Tensor &self, const at::Tensor &source, c10::SymInt storage_offset, c10::SymIntArrayRef size, c10::SymIntArrayRef stride={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | view (const at::Tensor &self, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | view (const at::Tensor &self, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | trace_backward (const at::Tensor &grad, at::IntArrayRef sizes) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | trace_backward (const at::Tensor &grad, c10::SymIntArrayRef sizes) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | index_select_backward (const at::Tensor &grad, at::IntArrayRef self_sizes, int64_t dim, const at::Tensor &index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | index_select_backward (const at::Tensor &grad, c10::SymIntArrayRef self_sizes, int64_t dim, const at::Tensor &index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | cross_entropy_loss (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, int64_t ignore_index=-100, double label_smoothing=0.0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | cross_entropy_loss (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, c10::SymInt ignore_index=-100, double label_smoothing=0.0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | unfold_backward (const at::Tensor &grad_in, at::IntArrayRef input_sizes, int64_t dim, int64_t size, int64_t step) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | unfold_backward (const at::Tensor &grad_in, c10::SymIntArrayRef input_sizes, int64_t dim, int64_t size, int64_t step) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | normal (double mean, double std, at::IntArrayRef size, c10::optional< at::Generator > generator=c10::nullopt, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | normal (double mean, double std, at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | normal (double mean, double std, c10::SymIntArrayRef size, c10::optional< at::Generator > generator=c10::nullopt, at::TensorOptions options={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | normal (double mean, double std, c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::ScalarType > dtype, c10::optional< at::Layout > layout, c10::optional< at::Device > device, c10::optional< bool > pin_memory) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | normal_out (at::Tensor &out, double mean, double std, at::IntArrayRef size, c10::optional< at::Generator > generator=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | normal_outf (double mean, double std, at::IntArrayRef size, c10::optional< at::Generator > generator, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | normal_out (at::Tensor &out, double mean, double std, c10::SymIntArrayRef size, c10::optional< at::Generator > generator=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | normal_outf (double mean, double std, c10::SymIntArrayRef size, c10::optional< at::Generator > generator, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | nll_loss_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, int64_t ignore_index=-100) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | nll_loss_outf (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | nll_loss_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, c10::SymInt ignore_index=-100) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | nll_loss_outf (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | nll_loss_nd (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, int64_t ignore_index=-100) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | nll_loss_nd (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, c10::SymInt ignore_index=-100) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | nll_loss (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, int64_t ignore_index=-100) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | nll_loss (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, c10::SymInt ignore_index=-100) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor & > | nll_loss_forward_out (at::Tensor &output, at::Tensor &total_weight, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor & > | nll_loss_forward_outf (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index, at::Tensor &output, at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor & > | nll_loss_forward_out (at::Tensor &output, at::Tensor &total_weight, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor & > | nll_loss_forward_outf (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index, at::Tensor &output, at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor, at::Tensor > | nll_loss_forward (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor, at::Tensor > | nll_loss_forward (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | nll_loss_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index, const at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | nll_loss_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index, const at::Tensor &total_weight, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | nll_loss_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index, const at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | nll_loss_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index, const at::Tensor &total_weight, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | nll_loss_backward (const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index, const at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | nll_loss_backward (const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index, const at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | nll_loss2d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, int64_t ignore_index=-100) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | nll_loss2d_outf (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | nll_loss2d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, c10::SymInt ignore_index=-100) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | nll_loss2d_outf (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | nll_loss2d (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, int64_t ignore_index=-100) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | nll_loss2d (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight={}, int64_t reduction=at::Reduction::Mean, c10::SymInt ignore_index=-100) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor & > | nll_loss2d_forward_out (at::Tensor &output, at::Tensor &total_weight, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor & > | nll_loss2d_forward_outf (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index, at::Tensor &output, at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor & > | nll_loss2d_forward_out (at::Tensor &output, at::Tensor &total_weight, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor & > | nll_loss2d_forward_outf (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index, at::Tensor &output, at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor, at::Tensor > | nll_loss2d_forward (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor, at::Tensor > | nll_loss2d_forward (const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | nll_loss2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index, const at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | nll_loss2d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index, const at::Tensor &total_weight, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | nll_loss2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index, const at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | nll_loss2d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index, const at::Tensor &total_weight, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | nll_loss2d_backward (const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, int64_t ignore_index, const at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | nll_loss2d_backward (const at::Tensor &grad_output, const at::Tensor &self, const at::Tensor &target, const c10::optional< at::Tensor > &weight, int64_t reduction, c10::SymInt ignore_index, const at::Tensor &total_weight) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | adaptive_avg_pool2d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | adaptive_avg_pool2d_outf (const at::Tensor &self, at::IntArrayRef output_size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | adaptive_avg_pool2d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | adaptive_avg_pool2d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | adaptive_avg_pool2d (const at::Tensor &self, at::IntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | adaptive_avg_pool2d (const at::Tensor &self, c10::SymIntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _adaptive_avg_pool2d (const at::Tensor &self, at::IntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _adaptive_avg_pool2d (const at::Tensor &self, c10::SymIntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | adaptive_avg_pool3d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | adaptive_avg_pool3d_outf (const at::Tensor &self, at::IntArrayRef output_size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | adaptive_avg_pool3d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | adaptive_avg_pool3d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | adaptive_avg_pool3d (const at::Tensor &self, at::IntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | adaptive_avg_pool3d (const at::Tensor &self, c10::SymIntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _adaptive_avg_pool3d (const at::Tensor &self, at::IntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _adaptive_avg_pool3d (const at::Tensor &self, c10::SymIntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad1d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad1d_outf (const at::Tensor &self, at::IntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad1d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad1d_outf (const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | reflection_pad1d (const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | reflection_pad1d (const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad1d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad1d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad1d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad1d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | reflection_pad1d_backward (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | reflection_pad1d_backward (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad2d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad2d_outf (const at::Tensor &self, at::IntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad2d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad2d_outf (const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | reflection_pad2d (const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | reflection_pad2d (const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad2d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad2d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | reflection_pad2d_backward (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | reflection_pad2d_backward (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad3d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad3d_outf (const at::Tensor &self, at::IntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad3d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad3d_outf (const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | reflection_pad3d (const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | reflection_pad3d (const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad3d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | reflection_pad3d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad3d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | reflection_pad3d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | reflection_pad3d_backward (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | reflection_pad3d_backward (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad1d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad1d_outf (const at::Tensor &self, at::IntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad1d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad1d_outf (const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | replication_pad1d (const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | replication_pad1d (const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad1d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad1d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad1d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad1d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | replication_pad1d_backward (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | replication_pad1d_backward (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad2d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad2d_outf (const at::Tensor &self, at::IntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad2d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad2d_outf (const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | replication_pad2d (const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | replication_pad2d (const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad2d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad2d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | replication_pad2d_backward (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | replication_pad2d_backward (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad3d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad3d_outf (const at::Tensor &self, at::IntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad3d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad3d_outf (const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | replication_pad3d (const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | replication_pad3d (const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad3d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | replication_pad3d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad3d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | replication_pad3d_backward_outf (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | replication_pad3d_backward (const at::Tensor &grad_output, const at::Tensor &self, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | replication_pad3d_backward (const at::Tensor &grad_output, const at::Tensor &self, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _pad_circular (const at::Tensor &self, at::IntArrayRef pad) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _pad_circular (const at::Tensor &self, c10::SymIntArrayRef pad) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _pad_enum (const at::Tensor &self, at::IntArrayRef pad, int64_t mode, c10::optional< double > value=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _pad_enum (const at::Tensor &self, c10::SymIntArrayRef pad, int64_t mode, c10::optional< double > value=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | pad (const at::Tensor &self, at::IntArrayRef pad, c10::string_view mode="constant", c10::optional< double > value=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | pad (const at::Tensor &self, c10::SymIntArrayRef pad, c10::string_view mode="constant", c10::optional< double > value=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_linear1d (const at::Tensor &input, at::OptionalIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_linear1d (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_bilinear2d (const at::Tensor &input, at::OptionalIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_bilinear2d (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_bilinear2d_aa (const at::Tensor &input, at::OptionalIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_bilinear2d_aa (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_trilinear3d (const at::Tensor &input, at::OptionalIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_trilinear3d (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_bicubic2d (const at::Tensor &input, at::OptionalIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_bicubic2d (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_bicubic2d_aa (const at::Tensor &input, at::OptionalIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_bicubic2d_aa (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, bool align_corners, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_nearest1d (const at::Tensor &input, at::OptionalIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_nearest1d (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_nearest_exact1d (const at::Tensor &input, at::OptionalIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_nearest_exact1d (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_nearest2d (const at::Tensor &input, at::OptionalIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_nearest2d (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_nearest_exact2d (const at::Tensor &input, at::OptionalIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_nearest_exact2d (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_nearest3d (const at::Tensor &input, at::OptionalIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_nearest3d (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_nearest_exact3d (const at::Tensor &input, at::OptionalIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_nearest_exact3d (const at::Tensor &input, at::OptionalSymIntArrayRef output_size, c10::optional< at::ArrayRef< double > > scale_factors) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_linear1d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_linear1d_outf (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_linear1d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_linear1d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_linear1d (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_linear1d (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_linear1d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_linear1d_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_linear1d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_linear1d_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_linear1d_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_linear1d_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_bilinear2d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_bilinear2d_outf (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_bilinear2d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_bilinear2d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_bilinear2d (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_bilinear2d (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_bilinear2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_bilinear2d_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_bilinear2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_bilinear2d_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_bilinear2d_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_bilinear2d_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_bilinear2d_aa_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_bilinear2d_aa_outf (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_bilinear2d_aa_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_bilinear2d_aa_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_bilinear2d_aa (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_bilinear2d_aa (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_bilinear2d_aa_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_bilinear2d_aa_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_bilinear2d_aa_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_bilinear2d_aa_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_bilinear2d_aa_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_bilinear2d_aa_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_bicubic2d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_bicubic2d_outf (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_bicubic2d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_bicubic2d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_bicubic2d (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_bicubic2d (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_bicubic2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_bicubic2d_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_bicubic2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_bicubic2d_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_bicubic2d_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_bicubic2d_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_bicubic2d_aa_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_bicubic2d_aa_outf (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_bicubic2d_aa_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_bicubic2d_aa_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_bicubic2d_aa (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_bicubic2d_aa (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_bicubic2d_aa_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_bicubic2d_aa_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_bicubic2d_aa_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_bicubic2d_aa_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_bicubic2d_aa_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_bicubic2d_aa_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_trilinear3d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_trilinear3d_outf (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_trilinear3d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_trilinear3d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_trilinear3d (const at::Tensor &self, at::IntArrayRef output_size, bool align_corners, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_trilinear3d (const at::Tensor &self, c10::SymIntArrayRef output_size, bool align_corners, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_trilinear3d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_trilinear3d_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_trilinear3d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_trilinear3d_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_trilinear3d_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, bool align_corners, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_trilinear3d_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, bool align_corners, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest1d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest1d_outf (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest1d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest1d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact1d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact1d_outf (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact1d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact1d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_nearest1d (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_nearest1d (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_nearest_exact1d (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_nearest_exact1d (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest1d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest1d_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest1d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest1d_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact1d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact1d_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact1d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact1d_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_nearest1d_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_nearest1d_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_nearest_exact1d_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_nearest_exact1d_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest2d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest2d_outf (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest2d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest2d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact2d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact2d_outf (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact2d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact2d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_nearest2d (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_nearest2d (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_nearest_exact2d (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_nearest_exact2d (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest2d_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest2d_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact2d_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact2d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact2d_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_nearest2d_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_nearest2d_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_nearest_exact2d_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_nearest_exact2d_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest3d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest3d_outf (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest3d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest3d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact3d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact3d_outf (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact3d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact3d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_nearest3d (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_nearest3d (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_nearest_exact3d (const at::Tensor &self, at::IntArrayRef output_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_nearest_exact3d (const at::Tensor &self, c10::SymIntArrayRef output_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest3d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | upsample_nearest3d_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest3d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | upsample_nearest3d_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact3d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _upsample_nearest_exact3d_backward_outf (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact3d_backward_out (at::Tensor &grad_input, const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _upsample_nearest_exact3d_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_d, c10::optional< double > scales_h, c10::optional< double > scales_w, at::Tensor &grad_input) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | upsample_nearest3d_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | upsample_nearest3d_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _upsample_nearest_exact3d_backward (const at::Tensor &grad_output, at::IntArrayRef output_size, at::IntArrayRef input_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _upsample_nearest_exact3d_backward (const at::Tensor &grad_output, c10::SymIntArrayRef output_size, c10::SymIntArrayRef input_size, c10::optional< double > scales_d=c10::nullopt, c10::optional< double > scales_h=c10::nullopt, c10::optional< double > scales_w=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv_transpose2d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, at::IntArrayRef padding=0, at::IntArrayRef output_padding=0, at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv_transpose2d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef output_padding, at::IntArrayRef dilation, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv_transpose2d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, c10::SymIntArrayRef padding=c10::SymInt(0), c10::SymIntArrayRef output_padding=c10::SymInt(0), at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv_transpose2d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, c10::SymIntArrayRef output_padding, at::IntArrayRef dilation, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | slow_conv_transpose2d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, at::IntArrayRef padding=0, at::IntArrayRef output_padding=0, at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | slow_conv_transpose2d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, c10::SymIntArrayRef padding=c10::SymInt(0), c10::SymIntArrayRef output_padding=c10::SymInt(0), at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv_transpose3d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, at::IntArrayRef padding=0, at::IntArrayRef output_padding=0, at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv_transpose3d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef output_padding, at::IntArrayRef dilation, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv_transpose3d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, c10::SymIntArrayRef padding=c10::SymInt(0), c10::SymIntArrayRef output_padding=c10::SymInt(0), at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv_transpose3d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, c10::SymIntArrayRef output_padding, at::IntArrayRef dilation, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | slow_conv_transpose3d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, at::IntArrayRef padding=0, at::IntArrayRef output_padding=0, at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | slow_conv_transpose3d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, c10::SymIntArrayRef padding=c10::SymInt(0), c10::SymIntArrayRef output_padding=c10::SymInt(0), at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
const at::Tensor & | _conv_depthwise2d_out (const at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
const at::Tensor & | _conv_depthwise2d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, const at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
const at::Tensor & | _conv_depthwise2d_out (const at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
const at::Tensor & | _conv_depthwise2d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, const at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _conv_depthwise2d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _conv_depthwise2d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | conv_depthwise3d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | conv_depthwise3d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv3d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, at::IntArrayRef padding=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv3d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv3d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, c10::SymIntArrayRef padding=c10::SymInt(0)) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv3d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | slow_conv3d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, at::IntArrayRef padding=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | slow_conv3d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, c10::SymIntArrayRef padding=c10::SymInt(0)) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv3d_forward_out (at::Tensor &output, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv3d_forward_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::Tensor &output) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv3d_forward_out (at::Tensor &output, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv3d_forward_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::Tensor &output) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | slow_conv3d_forward (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | slow_conv3d_forward (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | slow_conv_dilated2d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, at::IntArrayRef padding=0, at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | slow_conv_dilated2d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, c10::SymIntArrayRef padding=c10::SymInt(0), at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | slow_conv_dilated3d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, at::IntArrayRef padding=0, at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | slow_conv_dilated3d (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, c10::SymIntArrayRef padding=c10::SymInt(0), at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | col2im_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size, at::IntArrayRef kernel_size, at::IntArrayRef dilation, at::IntArrayRef padding, at::IntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | col2im_outf (const at::Tensor &self, at::IntArrayRef output_size, at::IntArrayRef kernel_size, at::IntArrayRef dilation, at::IntArrayRef padding, at::IntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | col2im_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size, at::IntArrayRef kernel_size, at::IntArrayRef dilation, at::IntArrayRef padding, at::IntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | col2im_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, at::IntArrayRef kernel_size, at::IntArrayRef dilation, at::IntArrayRef padding, at::IntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | col2im (const at::Tensor &self, at::IntArrayRef output_size, at::IntArrayRef kernel_size, at::IntArrayRef dilation, at::IntArrayRef padding, at::IntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | col2im (const at::Tensor &self, c10::SymIntArrayRef output_size, at::IntArrayRef kernel_size, at::IntArrayRef dilation, at::IntArrayRef padding, at::IntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | as_strided_copy (const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride, c10::optional< int64_t > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | as_strided_copy (const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, c10::optional< c10::SymInt > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | expand_copy (const at::Tensor &self, at::IntArrayRef size, bool implicit=false) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | expand_copy (const at::Tensor &self, c10::SymIntArrayRef size, bool implicit=false) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | _reshape_alias_copy (const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | _reshape_alias_copy (const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | select_copy (const at::Tensor &self, int64_t dim, int64_t index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | select_copy (const at::Tensor &self, int64_t dim, c10::SymInt index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | slice_copy (const at::Tensor &self, int64_t dim=0, c10::optional< int64_t > start=c10::nullopt, c10::optional< int64_t > end=c10::nullopt, int64_t step=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | slice_copy (const at::Tensor &self, int64_t dim=0, c10::optional< c10::SymInt > start=c10::nullopt, c10::optional< c10::SymInt > end=c10::nullopt, c10::SymInt step=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::vector< at::Tensor > | split_copy (const at::Tensor &self, int64_t split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::vector< at::Tensor > | split_copy (const at::Tensor &self, c10::SymInt split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::vector< at::Tensor > | split_with_sizes_copy (const at::Tensor &self, at::IntArrayRef split_sizes, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::vector< at::Tensor > | split_with_sizes_copy (const at::Tensor &self, c10::SymIntArrayRef split_sizes, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
void | split_copy_out (at::TensorList out, const at::Tensor &self, int64_t split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
void | split_copy_outf (const at::Tensor &self, int64_t split_size, int64_t dim, at::TensorList out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
void | split_copy_out (at::TensorList out, const at::Tensor &self, c10::SymInt split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
void | split_copy_outf (const at::Tensor &self, c10::SymInt split_size, int64_t dim, at::TensorList out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
void | split_with_sizes_copy_out (at::TensorList out, const at::Tensor &self, at::IntArrayRef split_sizes, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
void | split_with_sizes_copy_outf (const at::Tensor &self, at::IntArrayRef split_sizes, int64_t dim, at::TensorList out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
void | split_with_sizes_copy_out (at::TensorList out, const at::Tensor &self, c10::SymIntArrayRef split_sizes, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
void | split_with_sizes_copy_outf (const at::Tensor &self, c10::SymIntArrayRef split_sizes, int64_t dim, at::TensorList out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | view_copy (const at::Tensor &self, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | view_copy (const at::Tensor &self, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | to_padded_tensor (const at::Tensor &self, double padding, at::OptionalIntArrayRef output_size=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | to_padded_tensor (const at::Tensor &self, double padding, at::OptionalSymIntArrayRef output_size=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _cudnn_rnn_flatten_weight_out (at::Tensor &out, at::TensorList weight_arr, int64_t weight_stride0, int64_t input_size, int64_t mode, int64_t hidden_size, int64_t proj_size, int64_t num_layers, bool batch_first, bool bidirectional) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _cudnn_rnn_flatten_weight_outf (at::TensorList weight_arr, int64_t weight_stride0, int64_t input_size, int64_t mode, int64_t hidden_size, int64_t proj_size, int64_t num_layers, bool batch_first, bool bidirectional, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _cudnn_rnn_flatten_weight_out (at::Tensor &out, at::TensorList weight_arr, int64_t weight_stride0, c10::SymInt input_size, int64_t mode, c10::SymInt hidden_size, c10::SymInt proj_size, int64_t num_layers, bool batch_first, bool bidirectional) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _cudnn_rnn_flatten_weight_outf (at::TensorList weight_arr, int64_t weight_stride0, c10::SymInt input_size, int64_t mode, c10::SymInt hidden_size, c10::SymInt proj_size, int64_t num_layers, bool batch_first, bool bidirectional, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor & > | _cudnn_rnn_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, at::Tensor &out3, at::Tensor &out4, const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const c10::optional< at::Tensor > &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, int64_t mode, int64_t hidden_size, int64_t proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, at::IntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor & > | _cudnn_rnn_outf (const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const c10::optional< at::Tensor > &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, int64_t mode, int64_t hidden_size, int64_t proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, at::IntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, at::Tensor &out3, at::Tensor &out4) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor & > | _cudnn_rnn_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, at::Tensor &out3, at::Tensor &out4, const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const c10::optional< at::Tensor > &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, int64_t mode, c10::SymInt hidden_size, c10::SymInt proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, c10::SymIntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor &, at::Tensor & > | _cudnn_rnn_outf (const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const c10::optional< at::Tensor > &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, int64_t mode, c10::SymInt hidden_size, c10::SymInt proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, c10::SymIntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, at::Tensor &out3, at::Tensor &out4) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
void | _cudnn_rnn_backward_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, at::TensorList out3, const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const at::Tensor &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, const at::Tensor &output, const c10::optional< at::Tensor > &grad_output, const c10::optional< at::Tensor > &grad_hy, const c10::optional< at::Tensor > &grad_cy, int64_t mode, int64_t hidden_size, int64_t proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, at::IntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state, const at::Tensor &reserve, ::std::array< bool, 4 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
void | _cudnn_rnn_backward_outf (const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const at::Tensor &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, const at::Tensor &output, const c10::optional< at::Tensor > &grad_output, const c10::optional< at::Tensor > &grad_hy, const c10::optional< at::Tensor > &grad_cy, int64_t mode, int64_t hidden_size, int64_t proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, at::IntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state, const at::Tensor &reserve, ::std::array< bool, 4 > output_mask, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, at::TensorList out3) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
void | _cudnn_rnn_backward_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, at::TensorList out3, const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const at::Tensor &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, const at::Tensor &output, const c10::optional< at::Tensor > &grad_output, const c10::optional< at::Tensor > &grad_hy, const c10::optional< at::Tensor > &grad_cy, int64_t mode, c10::SymInt hidden_size, c10::SymInt proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, c10::SymIntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state, const at::Tensor &reserve, ::std::array< bool, 4 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
void | _cudnn_rnn_backward_outf (const at::Tensor &input, at::TensorList weight, int64_t weight_stride0, const at::Tensor &weight_buf, const at::Tensor &hx, const c10::optional< at::Tensor > &cx, const at::Tensor &output, const c10::optional< at::Tensor > &grad_output, const c10::optional< at::Tensor > &grad_hy, const c10::optional< at::Tensor > &grad_cy, int64_t mode, c10::SymInt hidden_size, c10::SymInt proj_size, int64_t num_layers, bool batch_first, double dropout, bool train, bool bidirectional, c10::SymIntArrayRef batch_sizes, const c10::optional< at::Tensor > &dropout_state, const at::Tensor &reserve, ::std::array< bool, 4 > output_mask, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, at::TensorList out3) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | constant_pad_nd_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef pad, const at::Scalar &value=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | constant_pad_nd_outf (const at::Tensor &self, at::IntArrayRef pad, const at::Scalar &value, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | constant_pad_nd_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef pad, const at::Scalar &value=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | constant_pad_nd_outf (const at::Tensor &self, c10::SymIntArrayRef pad, const at::Scalar &value, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | convolution_out (at::Tensor &out, const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool transposed, at::IntArrayRef output_padding, int64_t groups) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | convolution_outf (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool transposed, at::IntArrayRef output_padding, int64_t groups, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | convolution_out (at::Tensor &out, const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, bool transposed, c10::SymIntArrayRef output_padding, int64_t groups) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | convolution_outf (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, bool transposed, c10::SymIntArrayRef output_padding, int64_t groups, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | convolution_backward_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, const at::Tensor &grad_output, const at::Tensor &input, const at::Tensor &weight, at::OptionalIntArrayRef bias_sizes, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool transposed, at::IntArrayRef output_padding, int64_t groups, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | convolution_backward_outf (const at::Tensor &grad_output, const at::Tensor &input, const at::Tensor &weight, at::OptionalIntArrayRef bias_sizes, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool transposed, at::IntArrayRef output_padding, int64_t groups, ::std::array< bool, 3 > output_mask, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | convolution_backward_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, const at::Tensor &grad_output, const at::Tensor &input, const at::Tensor &weight, at::OptionalSymIntArrayRef bias_sizes, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, bool transposed, c10::SymIntArrayRef output_padding, int64_t groups, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | convolution_backward_outf (const at::Tensor &grad_output, const at::Tensor &input, const at::Tensor &weight, at::OptionalSymIntArrayRef bias_sizes, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, bool transposed, c10::SymIntArrayRef output_padding, int64_t groups, ::std::array< bool, 3 > output_mask, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _convolution_out (at::Tensor &out, const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool transposed, at::IntArrayRef output_padding, int64_t groups, bool benchmark, bool deterministic, bool cudnn_enabled, bool allow_tf32) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _convolution_outf (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, bool transposed, at::IntArrayRef output_padding, int64_t groups, bool benchmark, bool deterministic, bool cudnn_enabled, bool allow_tf32, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _convolution_out (at::Tensor &out, const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, bool transposed, c10::SymIntArrayRef output_padding, int64_t groups, bool benchmark, bool deterministic, bool cudnn_enabled, bool allow_tf32) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _convolution_outf (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, bool transposed, c10::SymIntArrayRef output_padding, int64_t groups, bool benchmark, bool deterministic, bool cudnn_enabled, bool allow_tf32, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | diagonal_backward_out (at::Tensor &out, const at::Tensor &grad_output, at::IntArrayRef input_sizes, int64_t offset, int64_t dim1, int64_t dim2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | diagonal_backward_outf (const at::Tensor &grad_output, at::IntArrayRef input_sizes, int64_t offset, int64_t dim1, int64_t dim2, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | diagonal_backward_out (at::Tensor &out, const at::Tensor &grad_output, c10::SymIntArrayRef input_sizes, int64_t offset, int64_t dim1, int64_t dim2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | diagonal_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef input_sizes, int64_t offset, int64_t dim1, int64_t dim2, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | embedding_out (at::Tensor &out, const at::Tensor &weight, const at::Tensor &indices, int64_t padding_idx=-1, bool scale_grad_by_freq=false, bool sparse=false) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | embedding_outf (const at::Tensor &weight, const at::Tensor &indices, int64_t padding_idx, bool scale_grad_by_freq, bool sparse, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | embedding_out (at::Tensor &out, const at::Tensor &weight, const at::Tensor &indices, c10::SymInt padding_idx=-1, bool scale_grad_by_freq=false, bool sparse=false) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | embedding_outf (const at::Tensor &weight, const at::Tensor &indices, c10::SymInt padding_idx, bool scale_grad_by_freq, bool sparse, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | embedding_dense_backward_out (at::Tensor &out, const at::Tensor &grad_output, const at::Tensor &indices, int64_t num_weights, int64_t padding_idx, bool scale_grad_by_freq) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | embedding_dense_backward_outf (const at::Tensor &grad_output, const at::Tensor &indices, int64_t num_weights, int64_t padding_idx, bool scale_grad_by_freq, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | embedding_dense_backward_out (at::Tensor &out, const at::Tensor &grad_output, const at::Tensor &indices, c10::SymInt num_weights, c10::SymInt padding_idx, bool scale_grad_by_freq) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | embedding_dense_backward_outf (const at::Tensor &grad_output, const at::Tensor &indices, c10::SymInt num_weights, c10::SymInt padding_idx, bool scale_grad_by_freq, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _embedding_bag_dense_backward_out (at::Tensor &out, const at::Tensor &grad, const at::Tensor &indices, const at::Tensor &offset2bag, const at::Tensor &bag_size, const at::Tensor &maximum_indices, int64_t num_weights, bool scale_grad_by_freq, int64_t mode, const c10::optional< at::Tensor > &per_sample_weights, int64_t padding_idx=-1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _embedding_bag_dense_backward_outf (const at::Tensor &grad, const at::Tensor &indices, const at::Tensor &offset2bag, const at::Tensor &bag_size, const at::Tensor &maximum_indices, int64_t num_weights, bool scale_grad_by_freq, int64_t mode, const c10::optional< at::Tensor > &per_sample_weights, int64_t padding_idx, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _embedding_bag_dense_backward_out (at::Tensor &out, const at::Tensor &grad, const at::Tensor &indices, const at::Tensor &offset2bag, const at::Tensor &bag_size, const at::Tensor &maximum_indices, c10::SymInt num_weights, bool scale_grad_by_freq, int64_t mode, const c10::optional< at::Tensor > &per_sample_weights, int64_t padding_idx=-1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _embedding_bag_dense_backward_outf (const at::Tensor &grad, const at::Tensor &indices, const at::Tensor &offset2bag, const at::Tensor &bag_size, const at::Tensor &maximum_indices, c10::SymInt num_weights, bool scale_grad_by_freq, int64_t mode, const c10::optional< at::Tensor > &per_sample_weights, int64_t padding_idx, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | new_empty_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | new_empty_outf (const at::Tensor &self, at::IntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | new_empty_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | new_empty_outf (const at::Tensor &self, c10::SymIntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | new_empty_strided_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | new_empty_strided_outf (const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | new_empty_strided_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | new_empty_strided_outf (const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | new_full_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef size, const at::Scalar &fill_value) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | new_full_outf (const at::Tensor &self, at::IntArrayRef size, const at::Scalar &fill_value, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | new_full_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef size, const at::Scalar &fill_value) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | new_full_outf (const at::Tensor &self, c10::SymIntArrayRef size, const at::Scalar &fill_value, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | new_zeros_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | new_zeros_outf (const at::Tensor &self, at::IntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | new_zeros_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | new_zeros_outf (const at::Tensor &self, c10::SymIntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | new_ones_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | new_ones_outf (const at::Tensor &self, at::IntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | new_ones_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | new_ones_outf (const at::Tensor &self, c10::SymIntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
const at::Tensor & | resize_out (const at::Tensor &out, const at::Tensor &self, at::IntArrayRef size, c10::optional< at::MemoryFormat > memory_format=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
const at::Tensor & | resize_outf (const at::Tensor &self, at::IntArrayRef size, c10::optional< at::MemoryFormat > memory_format, const at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
const at::Tensor & | resize_out (const at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef size, c10::optional< at::MemoryFormat > memory_format=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
const at::Tensor & | resize_outf (const at::Tensor &self, c10::SymIntArrayRef size, c10::optional< at::MemoryFormat > memory_format, const at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | resize (const at::Tensor &self, at::IntArrayRef size, c10::optional< at::MemoryFormat > memory_format=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | resize (const at::Tensor &self, c10::SymIntArrayRef size, c10::optional< at::MemoryFormat > memory_format=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | empty_strided_out (at::Tensor &out, at::IntArrayRef size, at::IntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | empty_strided_outf (at::IntArrayRef size, at::IntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | empty_strided_out (at::Tensor &out, c10::SymIntArrayRef size, c10::SymIntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | empty_strided_outf (c10::SymIntArrayRef size, c10::SymIntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_group_norm_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, const at::Tensor &input, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, int64_t N, int64_t C, int64_t HxW, int64_t group, double eps) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_group_norm_outf (const at::Tensor &input, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, int64_t N, int64_t C, int64_t HxW, int64_t group, double eps, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_group_norm_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, const at::Tensor &input, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, c10::SymInt N, c10::SymInt C, c10::SymInt HxW, int64_t group, double eps) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_group_norm_outf (const at::Tensor &input, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, c10::SymInt N, c10::SymInt C, c10::SymInt HxW, int64_t group, double eps, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_group_norm_backward_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, const at::Tensor &grad_out, const at::Tensor &input, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, int64_t N, int64_t C, int64_t HxW, int64_t group, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_group_norm_backward_outf (const at::Tensor &grad_out, const at::Tensor &input, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, int64_t N, int64_t C, int64_t HxW, int64_t group, ::std::array< bool, 3 > output_mask, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_group_norm_backward_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, const at::Tensor &grad_out, const at::Tensor &input, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, c10::SymInt N, c10::SymInt C, c10::SymInt HxW, int64_t group, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_group_norm_backward_outf (const at::Tensor &grad_out, const at::Tensor &input, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, c10::SymInt N, c10::SymInt C, c10::SymInt HxW, int64_t group, ::std::array< bool, 3 > output_mask, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_layer_norm_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, const at::Tensor &input, at::IntArrayRef normalized_shape, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, double eps) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_layer_norm_outf (const at::Tensor &input, at::IntArrayRef normalized_shape, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, double eps, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_layer_norm_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, const at::Tensor &input, c10::SymIntArrayRef normalized_shape, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, double eps) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_layer_norm_outf (const at::Tensor &input, c10::SymIntArrayRef normalized_shape, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, double eps, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_layer_norm_backward_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, const at::Tensor &grad_out, const at::Tensor &input, at::IntArrayRef normalized_shape, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_layer_norm_backward_outf (const at::Tensor &grad_out, const at::Tensor &input, at::IntArrayRef normalized_shape, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, ::std::array< bool, 3 > output_mask, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_layer_norm_backward_out (at::Tensor &out0, at::Tensor &out1, at::Tensor &out2, const at::Tensor &grad_out, const at::Tensor &input, c10::SymIntArrayRef normalized_shape, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, ::std::array< bool, 3 > output_mask) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
::std::tuple< at::Tensor &, at::Tensor &, at::Tensor & > | native_layer_norm_backward_outf (const at::Tensor &grad_out, const at::Tensor &input, c10::SymIntArrayRef normalized_shape, const at::Tensor &mean, const at::Tensor &rstd, const c10::optional< at::Tensor > &weight, const c10::optional< at::Tensor > &bias, ::std::array< bool, 3 > output_mask, at::Tensor &out0, at::Tensor &out1, at::Tensor &out2) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | mkldnn_convolution_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | mkldnn_convolution_outf (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | mkldnn_convolution_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | mkldnn_convolution_outf (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | miopen_convolution_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | miopen_convolution_outf (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | miopen_convolution_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | miopen_convolution_outf (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | miopen_convolution_transpose_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef output_padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | miopen_convolution_transpose_outf (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef output_padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | miopen_convolution_transpose_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, c10::SymIntArrayRef output_padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | miopen_convolution_transpose_outf (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, c10::SymIntArrayRef output_padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | miopen_depthwise_convolution_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | miopen_depthwise_convolution_outf (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | miopen_depthwise_convolution_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | miopen_depthwise_convolution_outf (const at::Tensor &self, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride, at::IntArrayRef dilation, int64_t groups, bool benchmark, bool deterministic, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _nnpack_spatial_convolution_out (at::Tensor &out, const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _nnpack_spatial_convolution_outf (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, at::IntArrayRef padding, at::IntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _nnpack_spatial_convolution_out (at::Tensor &out, const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _nnpack_spatial_convolution_outf (const at::Tensor &input, const at::Tensor &weight, const c10::optional< at::Tensor > &bias, c10::SymIntArrayRef padding, at::IntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | rand_out (at::Tensor &out, at::IntArrayRef size, c10::optional< at::DimnameList > names) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | rand_outf (at::IntArrayRef size, c10::optional< at::DimnameList > names, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | rand_out (at::Tensor &out, c10::SymIntArrayRef size, c10::optional< at::DimnameList > names) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | rand_outf (c10::SymIntArrayRef size, c10::optional< at::DimnameList > names, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | rand_out (at::Tensor &out, at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | rand_outf (at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | rand_out (at::Tensor &out, c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | rand_outf (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randn_out (at::Tensor &out, at::IntArrayRef size, c10::optional< at::DimnameList > names) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randn_outf (at::IntArrayRef size, c10::optional< at::DimnameList > names, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randn_out (at::Tensor &out, c10::SymIntArrayRef size, c10::optional< at::DimnameList > names) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randn_outf (c10::SymIntArrayRef size, c10::optional< at::DimnameList > names, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randn_out (at::Tensor &out, at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | randn_outf (at::IntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randn_out (at::Tensor &out, c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | randn_outf (c10::SymIntArrayRef size, c10::optional< at::Generator > generator, c10::optional< at::DimnameList > names, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | repeat_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef repeats) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | repeat_outf (const at::Tensor &self, at::IntArrayRef repeats, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | repeat_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef repeats) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | repeat_outf (const at::Tensor &self, c10::SymIntArrayRef repeats, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | select_backward_out (at::Tensor &out, const at::Tensor &grad_output, at::IntArrayRef input_sizes, int64_t dim, int64_t index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | select_backward_outf (const at::Tensor &grad_output, at::IntArrayRef input_sizes, int64_t dim, int64_t index, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | select_backward_out (at::Tensor &out, const at::Tensor &grad_output, c10::SymIntArrayRef input_sizes, int64_t dim, c10::SymInt index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | select_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef input_sizes, int64_t dim, c10::SymInt index, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slice_backward_out (at::Tensor &out, const at::Tensor &grad_output, at::IntArrayRef input_sizes, int64_t dim, int64_t start, int64_t end, int64_t step) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slice_backward_outf (const at::Tensor &grad_output, at::IntArrayRef input_sizes, int64_t dim, int64_t start, int64_t end, int64_t step, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slice_backward_out (at::Tensor &out, const at::Tensor &grad_output, c10::SymIntArrayRef input_sizes, int64_t dim, c10::SymInt start, c10::SymInt end, c10::SymInt step) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slice_backward_outf (const at::Tensor &grad_output, c10::SymIntArrayRef input_sizes, int64_t dim, c10::SymInt start, c10::SymInt end, c10::SymInt step, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slice_scatter_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &src, int64_t dim=0, c10::optional< int64_t > start=c10::nullopt, c10::optional< int64_t > end=c10::nullopt, int64_t step=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slice_scatter_outf (const at::Tensor &self, const at::Tensor &src, int64_t dim, c10::optional< int64_t > start, c10::optional< int64_t > end, int64_t step, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slice_scatter_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &src, int64_t dim=0, c10::optional< c10::SymInt > start=c10::nullopt, c10::optional< c10::SymInt > end=c10::nullopt, c10::SymInt step=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slice_scatter_outf (const at::Tensor &self, const at::Tensor &src, int64_t dim, c10::optional< c10::SymInt > start, c10::optional< c10::SymInt > end, c10::SymInt step, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | select_scatter_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &src, int64_t dim, int64_t index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | select_scatter_outf (const at::Tensor &self, const at::Tensor &src, int64_t dim, int64_t index, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | select_scatter_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &src, int64_t dim, c10::SymInt index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | select_scatter_outf (const at::Tensor &self, const at::Tensor &src, int64_t dim, c10::SymInt index, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | as_strided_scatter_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &src, at::IntArrayRef size, at::IntArrayRef stride, c10::optional< int64_t > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | as_strided_scatter_outf (const at::Tensor &self, const at::Tensor &src, at::IntArrayRef size, at::IntArrayRef stride, c10::optional< int64_t > storage_offset, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | as_strided_scatter_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &src, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, c10::optional< c10::SymInt > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | as_strided_scatter_outf (const at::Tensor &self, const at::Tensor &src, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, c10::optional< c10::SymInt > storage_offset, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
void | unsafe_split_out (at::TensorList out, const at::Tensor &self, int64_t split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
void | unsafe_split_outf (const at::Tensor &self, int64_t split_size, int64_t dim, at::TensorList out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
void | unsafe_split_out (at::TensorList out, const at::Tensor &self, c10::SymInt split_size, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
void | unsafe_split_outf (const at::Tensor &self, c10::SymInt split_size, int64_t dim, at::TensorList out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
void | unsafe_split_with_sizes_out (at::TensorList out, const at::Tensor &self, at::IntArrayRef split_sizes, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
void | unsafe_split_with_sizes_outf (const at::Tensor &self, at::IntArrayRef split_sizes, int64_t dim, at::TensorList out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
void | unsafe_split_with_sizes_out (at::TensorList out, const at::Tensor &self, c10::SymIntArrayRef split_sizes, int64_t dim=0) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
void | unsafe_split_with_sizes_outf (const at::Tensor &self, c10::SymIntArrayRef split_sizes, int64_t dim, at::TensorList out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _unsafe_view_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _unsafe_view_outf (const at::Tensor &self, at::IntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _unsafe_view_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _unsafe_view_outf (const at::Tensor &self, c10::SymIntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _sparse_coo_tensor_with_dims_and_tensors_out (at::Tensor &out, int64_t sparse_dim, int64_t dense_dim, at::IntArrayRef size, const at::Tensor &indices, const at::Tensor &values) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _sparse_coo_tensor_with_dims_and_tensors_outf (int64_t sparse_dim, int64_t dense_dim, at::IntArrayRef size, const at::Tensor &indices, const at::Tensor &values, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _sparse_coo_tensor_with_dims_and_tensors_out (at::Tensor &out, int64_t sparse_dim, int64_t dense_dim, c10::SymIntArrayRef size, const at::Tensor &indices, const at::Tensor &values) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _sparse_coo_tensor_with_dims_and_tensors_outf (int64_t sparse_dim, int64_t dense_dim, c10::SymIntArrayRef size, const at::Tensor &indices, const at::Tensor &values, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | set_out (at::Tensor &out, const at::Tensor &self, at::Storage source, int64_t storage_offset, at::IntArrayRef size, at::IntArrayRef stride={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | set_outf (const at::Tensor &self, at::Storage source, int64_t storage_offset, at::IntArrayRef size, at::IntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | set_out (at::Tensor &out, const at::Tensor &self, at::Storage source, c10::SymInt storage_offset, c10::SymIntArrayRef size, c10::SymIntArrayRef stride={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | set_outf (const at::Tensor &self, at::Storage source, c10::SymInt storage_offset, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor | set (const at::Tensor &self, at::Storage source, int64_t storage_offset, at::IntArrayRef size, at::IntArrayRef stride={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor | set (const at::Tensor &self, at::Storage source, c10::SymInt storage_offset, c10::SymIntArrayRef size, c10::SymIntArrayRef stride={}) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | unfold_backward_out (at::Tensor &out, const at::Tensor &grad_in, at::IntArrayRef input_sizes, int64_t dim, int64_t size, int64_t step) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | unfold_backward_outf (const at::Tensor &grad_in, at::IntArrayRef input_sizes, int64_t dim, int64_t size, int64_t step, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | unfold_backward_out (at::Tensor &out, const at::Tensor &grad_in, c10::SymIntArrayRef input_sizes, int64_t dim, int64_t size, int64_t step) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | unfold_backward_outf (const at::Tensor &grad_in, c10::SymIntArrayRef input_sizes, int64_t dim, int64_t size, int64_t step, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _adaptive_avg_pool2d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _adaptive_avg_pool2d_outf (const at::Tensor &self, at::IntArrayRef output_size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _adaptive_avg_pool2d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _adaptive_avg_pool2d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _adaptive_avg_pool3d_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _adaptive_avg_pool3d_outf (const at::Tensor &self, at::IntArrayRef output_size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _adaptive_avg_pool3d_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef output_size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _adaptive_avg_pool3d_outf (const at::Tensor &self, c10::SymIntArrayRef output_size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | conv_depthwise3d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | conv_depthwise3d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | conv_depthwise3d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | conv_depthwise3d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv_dilated2d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, at::IntArrayRef padding=0, at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv_dilated2d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv_dilated2d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, c10::SymIntArrayRef padding=c10::SymInt(0), at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv_dilated2d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv_dilated3d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, at::IntArrayRef padding=0, at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slow_conv_dilated3d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, at::IntArrayRef padding, at::IntArrayRef dilation, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv_dilated3d_out (at::Tensor &out, const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias={}, at::IntArrayRef stride=1, c10::SymIntArrayRef padding=c10::SymInt(0), at::IntArrayRef dilation=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slow_conv_dilated3d_outf (const at::Tensor &self, const at::Tensor &weight, at::IntArrayRef kernel_size, const c10::optional< at::Tensor > &bias, at::IntArrayRef stride, c10::SymIntArrayRef padding, at::IntArrayRef dilation, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | as_strided_copy_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride, c10::optional< int64_t > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | as_strided_copy_outf (const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride, c10::optional< int64_t > storage_offset, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | as_strided_copy_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, c10::optional< c10::SymInt > storage_offset=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | as_strided_copy_outf (const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, c10::optional< c10::SymInt > storage_offset, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | expand_copy_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef size, bool implicit=false) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | expand_copy_outf (const at::Tensor &self, at::IntArrayRef size, bool implicit, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | expand_copy_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef size, bool implicit=false) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | expand_copy_outf (const at::Tensor &self, c10::SymIntArrayRef size, bool implicit, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _reshape_alias_copy_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | _reshape_alias_copy_outf (const at::Tensor &self, at::IntArrayRef size, at::IntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _reshape_alias_copy_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | _reshape_alias_copy_outf (const at::Tensor &self, c10::SymIntArrayRef size, c10::SymIntArrayRef stride, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | select_copy_out (at::Tensor &out, const at::Tensor &self, int64_t dim, int64_t index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | select_copy_outf (const at::Tensor &self, int64_t dim, int64_t index, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | select_copy_out (at::Tensor &out, const at::Tensor &self, int64_t dim, c10::SymInt index) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | select_copy_outf (const at::Tensor &self, int64_t dim, c10::SymInt index, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slice_copy_out (at::Tensor &out, const at::Tensor &self, int64_t dim=0, c10::optional< int64_t > start=c10::nullopt, c10::optional< int64_t > end=c10::nullopt, int64_t step=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | slice_copy_outf (const at::Tensor &self, int64_t dim, c10::optional< int64_t > start, c10::optional< int64_t > end, int64_t step, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slice_copy_out (at::Tensor &out, const at::Tensor &self, int64_t dim=0, c10::optional< c10::SymInt > start=c10::nullopt, c10::optional< c10::SymInt > end=c10::nullopt, c10::SymInt step=1) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | slice_copy_outf (const at::Tensor &self, int64_t dim, c10::optional< c10::SymInt > start, c10::optional< c10::SymInt > end, c10::SymInt step, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | view_copy_out (at::Tensor &out, const at::Tensor &self, at::IntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | view_copy_outf (const at::Tensor &self, at::IntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | view_copy_out (at::Tensor &out, const at::Tensor &self, c10::SymIntArrayRef size) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | view_copy_outf (const at::Tensor &self, c10::SymIntArrayRef size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | to_padded_tensor_out (at::Tensor &out, const at::Tensor &self, double padding, at::OptionalIntArrayRef output_size=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, int64_t>::value>> |
at::Tensor & | to_padded_tensor_outf (const at::Tensor &self, double padding, at::OptionalIntArrayRef output_size, at::Tensor &out) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | to_padded_tensor_out (at::Tensor &out, const at::Tensor &self, double padding, at::OptionalSymIntArrayRef output_size=c10::nullopt) |
|
template<typename T , typename = std::enable_if_t<std::is_same<T, c10::SymInt>::value>> |
at::Tensor & | to_padded_tensor_outf (const at::Tensor &self, double padding, at::OptionalSymIntArrayRef output_size, at::Tensor &out) |
|