|
ScalarType | scalarType () const override |
|
caffe2::TypeMeta | typeMeta () const override |
|
Backend | backend () const override |
|
const char * | toString () const override |
|
TypeID | ID () const override |
|
Tensor | empty (IntArrayRef size, const TensorOptions &options) const override |
|
| CPUTypeDefault (TensorTypeId type_id, bool is_variable, bool is_undefined) |
|
Allocator * | allocator () const override |
|
Device | getDeviceFromPtr (void *data) const override |
|
std::unique_ptr< Generator > | generator () 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 |
|