|
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 |
|
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) |
|
| DEFINE_IS_SUBCLASS (TensorType) |
|
virtual std::string | python_str () const |
|
TypeKind | kind () const |
|
template<typename T > |
std::shared_ptr< T > | cast () |
|
template<typename T > |
std::shared_ptr< const T > | cast () const |
|
template<typename T > |
std::shared_ptr< T > | expect () |
|
template<typename T > |
std::shared_ptr< const T > | expect () 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 |
|
|
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 () |
|
template<typename... T> |
static DimensionedTensorTypePtr | create (T &&...all) |
|
static TensorTypePtr | create () |
|
static TensorTypePtr | get () |
|
|
static const TypeKind | Kind = TypeKind::CompleteTensorType |
|
static const TypeKind | Kind = TypeKind::DimensionedTensorType |
|
static const TypeKind | Kind = TypeKind::TensorType |
|
|
| 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) |
|
| TensorType (TypeKind kind=TypeKind::TensorType) |
|
| Type (TypeKind kind) |
|
at::ScalarType | scalar_type_ |
|
bool | requires_grad_ |
|
at::Device | device_ |
|
int64_t | dim_ |
|
Definition at line 397 of file jit_type.h.
The documentation for this struct was generated from the following file: