PyTorch
Loading...
Searching...
No Matches
Classes | Namespaces | Typedefs | Functions
TensorBase.h File Reference
#include <c10/core/Device.h>
#include <c10/core/Layout.h>
#include <c10/core/MemoryFormat.h>
#include <c10/core/ScalarType.h>
#include <c10/core/ScalarTypeToTypeMeta.h>
#include <c10/core/Storage.h>
#include <c10/core/TensorImpl.h>
#include <c10/core/TensorOptions.h>
#include <c10/core/UndefinedTensorImpl.h>
#include <c10/core/WrapDimMinimal.h>
#include <c10/util/Exception.h>
#include <c10/util/ExclusivelyOwnedTensorTraits.h>
#include <c10/util/MaybeOwned.h>
#include <c10/util/Optional.h>
#include <c10/util/intrusive_ptr.h>
#include <ATen/core/NamedTensor.h>
#include <ATen/core/QuantizerBase.h>
#include <c10/core/SymIntArrayRef.h>
#include <ATen/core/TensorAccessor.h>

Go to the source code of this file.

Classes

class  at::TensorBase
 
struct  at::TensorBase::unsafe_borrow_t
 
struct  c10::MaybeOwnedTraits< at::TensorBase >
 
struct  c10::ExclusivelyOwnedTraits< at::TensorBase >
 

Namespaces

namespace  c10
 
namespace  torch
 
namespace  torch::autograd
 
namespace  at
 
namespace  at::impl
 
namespace  at::detail
 
namespace  at::symint
 

Typedefs

template<typename T >
using at::symint::enable_if_symint = std::enable_if_t< std::is_same< T, c10::SymInt >::value >
 
template<typename T >
using at::symint::enable_if_int = std::enable_if_t< std::is_same< T, int64_t >::value >
 

Functions

const TensorBase & at::get_tensor_base (const Tensor &t)
 
bool at::impl::variable_excluded_from_dispatch ()
 
int64_t at::get_device (const TensorBase &self)
 
template<typename T , typename... Args>
TensorBase at::detail::make_tensor_base (Args &&... args)
 
static DispatchKey at::legacyExtractDispatchKey (const TensorBase &t)
 
c10::MaybeOwned< TensorBase > at::borrow_from_optional_tensor (const c10::optional< TensorBase > &opt)
 
template<typename T , typename = enable_if_symint<T>>
c10::SymIntArrayRef at::symint::sizes (const TensorBase &t)
 
template<typename T , typename = enable_if_int<T>>
IntArrayRef at::symint::sizes (const TensorBase &t)
 
template<typename T , typename = enable_if_symint<T>>
c10::SymInt at::symint::size (const TensorBase &t, int64_t dim)
 
template<typename T , typename = enable_if_int<T>>
int64_t at::symint::size (const TensorBase &t, int64_t dim)
 
template<typename T , typename = enable_if_symint<T>>
c10::SymIntArrayRef at::symint::strides (const TensorBase &t)
 
template<typename T , typename = enable_if_int<T>>
IntArrayRef at::symint::strides (const TensorBase &t)
 
template<typename T , typename = enable_if_symint<T>>
c10::SymInt at::symint::numel (const TensorBase &t)
 
template<typename T , typename = enable_if_int<T>>
int64_t at::symint::numel (const TensorBase &t)