|
|
| VariableType (Context *context, at::TypeExtendedInterface *baseType) |
| |
|
at::ScalarType | scalarType () const override |
| |
|
caffe2::TypeMeta | typeMeta () const override |
| |
|
at::Backend | backend () const override |
| |
|
at::Allocator * | allocator () const override |
| |
|
at::Device | getDeviceFromPtr (void *data) const override |
| |
|
Storage | storageFromBlob (void *data, int64_t size, const std::function< void(void *)> &deleter) const override |
| |
|
Storage | storageWithAllocator (int64_t size, at::Allocator *allocator) const override |
| |
|
std::unique_ptr< at::Generator > | generator () const override |
| |
|
const char * | toString () const override |
| |
|
at::TypeID | ID () const override |
| |
|
at::Type & | toBackend (at::Backend b) const override |
| |
|
at::Type & | toScalarType (at::ScalarType s) const override |
| |
|
Storage | unsafeStorageFromTH (void *th_pointer, bool retain) const override |
| |
|
at::Tensor | unsafeTensorFromTH (void *th_pointer, bool retain) const override |
| |
|
void | backward (Tensor &self, c10::optional< Tensor > gradient, bool keep_graph, bool create_graph) const override |
| |
|
void | set_data (Tensor &self, Tensor new_data) const override |
| |
|
| TypeDefault (TensorTypeId type_id, bool is_variable, bool is_undefined) |
| |
|
bool | is_cuda () const override |
| |
|
bool | is_hip () const override |
| |
|
bool | is_sparse () const override |
| |
|
bool | is_distributed () const override |
| |
|
Type & | toBackend (Backend b) const override |
| |
|
Type & | toScalarType (ScalarType s) const override |
| |
|
Tensor | copy (const Tensor &src, bool non_blocking=false, optional< Device > to_device={}) const override |
| |
|
Tensor & | copy_ (Tensor &self, const Tensor &src, bool non_blocking=false) const override |
| |
|
void | backward (Tensor &self, c10::optional< Tensor > gradient, bool keep_graph, bool create_graph) const override |
| |
|
void | set_data (Tensor &self, Tensor new_data) const override |
| |
|
Tensor | tensorFromBlob (void *data, IntArrayRef sizes, const std::function< void(void *)> &deleter=noop_deleter) const override |
| |
|
Tensor | tensorFromBlob (void *data, IntArrayRef sizes, IntArrayRef strides, const std::function< void(void *)> &deleter=noop_deleter) const override |
| |
|
Tensor | tensorWithAllocator (IntArrayRef sizes, Allocator *allocator) const override |
| |
|
Tensor | tensorWithAllocator (IntArrayRef sizes, IntArrayRef strides, Allocator *allocator) const override |
| |
|
Storage | storageFromBlob (void *data, int64_t size, const std::function< void(void *)> &deleter) const override |
| |
|
Storage | storageWithAllocator (int64_t size, Allocator *allocator) const override |
| |
|
Storage | unsafeStorageFromTH (void *th_pointer, bool retain) const override |
| |
|
Tensor | unsafeTensorFromTH (void *th_pointer, bool retain) const override |
| |
|
| TypeExtendedInterface (TensorTypeId type_id, bool is_variable, bool is_undefined) |
| |
|
virtual bool | operator== (const Type &rhs) const =0 |
| |
|
virtual bool | isSubtypeOf (const TypePtr rhs) const |
| |
|
virtual bool | isSubclass (const TypeKind kind) const =0 |
| |
|
virtual std::string | str () const =0 |
| |
|
virtual std::string | python_str () const |
| |
|
TypeKind | kind () const |
| |
|
virtual bool | requires_grad () 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 |
| |
|
|
static at::TypeExtendedInterface * | getVariableTypeFromBaseType (const at::Type &baseType) |
| |
|
static bool | isVariableType (const at::Type &type) |
| |
|
static std::vector< at::Type * > | allCUDATypes () |
| |
|
static std::vector< at::Type * > | allCPUTypes () |
| |
|
static Variable & | checked_cast_variable (Tensor &t, const char *name, int pos) |
| |
|
static at::Tensor & | unpack (Tensor &t, const char *name, int pos) |
| |
|
static const at::Tensor & | unpack (const Tensor &t, const char *name, int pos) |
| |
|
static at::SparseTensorRef | unpack (SparseTensorRef t, const char *name, int pos) |
| |
|
static at::Tensor | unpack_opt (const Tensor &t, const char *name, int pos) |
| |
|
static std::vector< at::Tensor > | unpack (at::TensorList tl, const char *name, int pos) |
| |
Definition at line 34 of file VariableType.h.
Public Member Functions inherited from at::TypeDefault
1.8.11
