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

Public Member Functions

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)
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
- Public Member Functions inherited from c10::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 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::DimensionedTensorType
- Static Public Attributes inherited from c10::TensorType
static const TypeKind Kind = TypeKind::TensorType

Protected Member Functions

 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

at::ScalarType scalar_type_
bool requires_grad_
at::Device device_
int64_t dim_

Detailed Description

Definition at line 322 of file jit_type.h.

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