 |
Caffe2 - C++ API
A deep learning, cross platform ML framework
|
|
Caffe2 - C++ API
A deep learning, cross platform ML framework
|
Public Member Functions | |
| 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 |
 Public Member Functions inherited from at::TypeExtendedInterface | |
| TypeExtendedInterface (TensorTypeId type_id, bool is_variable, bool is_undefined) | |
| Public Member Functions inherited from c10::Type | |
| 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 |
Additional Inherited Members | |
| Protected Member Functions inherited from c10::Type | |
| Type (TypeKind kind) | |
Definition at line 9 of file TypeDefault.h.
1.8.11
