Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions | Static Public Attributes
c10::CompleteTensorType Struct Reference
Inheritance diagram for c10::CompleteTensorType:
c10::DimensionedTensorType c10::TensorType c10::Type

Public Member Functions

const std::vector< int64_t > & sizes () const
 
const std::vector< int64_t > & strides () const
 
TypePtr withSizesStrides (at::IntArrayRef sizes, at::IntArrayRef strides) const
 
TypePtr withSizes (at::IntArrayRef sizes) const
 
CompleteTensorTypePtr contiguous () const
 
CompleteTensorTypePtr toScalarType (at::ScalarType type)
 
bool operator== (const Type &rhs) const override
 
bool isSubtypeOf (const TypePtr rhs) const override
 
bool isSubclass (const TypeKind kind) const override
 
std::string str () const override
 
bool numel () const
 
- Public Member Functions inherited from c10::DimensionedTensorType
at::ScalarType scalarType () const
 
at::Device device () const
 
int64_t dim () const
 
bool requires_grad () const override
 
DimensionedTensorTypePtr toScalarType (at::ScalarType type)
 
DimensionedTensorTypePtr withDim (size_t new_dim)
 
DimensionedTensorTypePtr withRequiresGrad (bool req)
 
- Public Member Functions inherited from c10::TensorType
 DEFINE_IS_SUBCLASS (TensorType)
 
- Public Member Functions inherited from c10::Type
virtual std::string python_str () const
 
TypeKind kind () const
 
template<typename T >
std::shared_ptr< Tcast ()
 
template<typename T >
std::shared_ptr< const Tcast () const
 
template<typename T >
std::shared_ptr< Texpect ()
 
template<typename T >
std::shared_ptr< const Texpect () const
 
virtual bool hasFreeVariables () const
 
virtual at::ArrayRef< TypePtr > containedTypes () const
 
TypePtr withContained (std::vector< TypePtr > contained_types)
 
virtual TypePtr createWithContained (std::vector< TypePtr > contained_types) const
 

Static Public Member Functions

template<typename... T>
static CompleteTensorTypePtr create (T &&...all)
 
static CompleteTensorTypePtr create (at::ScalarType scalar_type, at::Device device, at::IntArrayRef sizes)
 
static CompleteTensorTypePtr create (at::ScalarType scalar_type, at::Device device, at::IntArrayRef sizes, at::IntArrayRef strides)
 
static TypePtr fromNumberType (TypePtr typ)
 
static TypePtr fromBoolType ()
 
- Static Public Member Functions inherited from c10::DimensionedTensorType
template<typename... T>
static DimensionedTensorTypePtr create (T &&...all)
 
- Static Public Member Functions inherited from c10::TensorType
static TensorTypePtr create ()
 
static TensorTypePtr get ()
 

Static Public Attributes

static const TypeKind Kind = TypeKind::CompleteTensorType
 
- Static Public Attributes inherited from c10::DimensionedTensorType
static const TypeKind Kind = TypeKind::DimensionedTensorType
 
- Static Public Attributes inherited from c10::TensorType
static const TypeKind Kind = TypeKind::TensorType
 

Additional Inherited Members

- Protected Member Functions inherited from c10::DimensionedTensorType
 DimensionedTensorType (const at::Tensor &tensor, TypeKind kind=TypeKind::DimensionedTensorType)
 
 DimensionedTensorType (at::ScalarType scalar_type, at::Device device, int64_t dim, bool requires_grad=true, TypeKind kind=TypeKind::DimensionedTensorType)
 
- Protected Member Functions inherited from c10::TensorType
 TensorType (TypeKind kind=TypeKind::TensorType)
 
- Protected Member Functions inherited from c10::Type
 Type (TypeKind kind)
 
- Protected Attributes inherited from c10::DimensionedTensorType
at::ScalarType scalar_type_
 
bool requires_grad_
 
at::Device device_
 
int64_t dim_
 

Detailed Description

Definition at line 397 of file jit_type.h.


The documentation for this struct was generated from the following file: