|
| | Divide (const Tensor input, const Tensor other, Tensor out) |
| |
| | Divide (const Tensor input, const Tensor other, const std::optional< std::string > rounding_mode, Tensor out) |
| |
| | Divide (Tensor input, const Tensor other) |
| |
| | Divide (const Tensor input, const double other, Tensor out) |
| |
| | Divide (Tensor input, const Tensor other, const std::optional< std::string > rounding_mode) |
| |
| | Divide (const Tensor input, const double other, const std::optional< std::string > rounding_mode, Tensor out) |
| |
| virtual void | operator() (const Tensor input, const Tensor other, Tensor out) const =0 |
| |
| virtual void | operator() (const Tensor input, const Tensor other, const std::optional< std::string > rounding_mode, Tensor out) const =0 |
| |
| virtual void | operator() (Tensor input, const Tensor other) const =0 |
| |
| virtual void | operator() (const Tensor input, const double other, Tensor out) const =0 |
| |
| virtual void | operator() (Tensor input, const Tensor other, const std::optional< std::string > rounding_mode) const =0 |
| |
| virtual void | operator() (const Tensor input, const double other, const std::optional< std::string > rounding_mode, Tensor out) const =0 |
| |
| void | operator() (const Handle &handle, const Args &... args) |
| |
| void | operator() (const Args &... args) const |
| |
| void | operator() (const Handle &handle, const Args &... args) |
| |
| void | operator() (const Args &... args) const |
| |
| virtual | ~OperatorBase ()=default |
| |
| virtual std::size_t | workspace_size_in_bytes () const |
| |
| void | set_handle (const Handle &handle) |
| |
| void | set_config (const Config &config) |
| |
| void | set_stream (void *stream) |
| |
| void | set_workspace (void *workspace) |
| |
| void | set_workspace_size_in_bytes (std::size_t workspace_size_in_bytes) |
| |
| virtual | ~OperatorBase ()=default |
| |
| virtual std::size_t | workspace_size_in_bytes () const |
| |
| void | set_handle (const Handle &handle) |
| |
| void | set_config (const Config &config) |
| |
| void | set_stream (void *stream) |
| |
| void | set_workspace (void *workspace) |
| |
| void | set_workspace_size_in_bytes (std::size_t workspace_size_in_bytes) |
| |
|
| static void | clear_cache () |
| |
| static void | clear_cache () |
| |
| static std::unique_ptr< Operator > | Make (const Config &config, const Tensor tensor, Args &&... args) |
| |
| static std::unique_ptr< Operator > | Make (const Tensor tensor, Args &&... args) |
| |
| static std::unique_ptr< Operator > | Make (const Config &config, const std::vector< Tensor > tensors, Args &&... args) |
| |
| static std::unique_ptr< Operator > | Make (const std::vector< Tensor > tensors, Args &&... args) |
| |
| static std::unique_ptr< Operator > | Make (const Config &config, const Tensor tensor, Args &&... args) |
| |
| static std::unique_ptr< Operator > | Make (const Tensor tensor, Args &&... args) |
| |
| static std::unique_ptr< Operator > | Make (const Config &config, const std::vector< Tensor > tensors, Args &&... args) |
| |
| static std::unique_ptr< Operator > | Make (const std::vector< Tensor > tensors, Args &&... args) |
| |
| static void | Call (const Handle &handle, const Config &config, const Args &... args) |
| |
| static void | Call (const Tensor tensor, const Args &... args) |
| |
| static auto | Call (const TensorLike &tensor, const Args &... args) |
| |
| static void | Call (const Handle &handle, const Config &config, const Args &... args) |
| |
| static void | Call (const Tensor tensor, const Args &... args) |
| |
| static auto | Call (const TensorLike &tensor, const Args &... args) |
| |
| static std::vector< std::size_t > | active_implementation_indices (Device::Type dev_type) |
| |
| static std::vector< std::size_t > | active_implementation_indices (Device::Type dev_type) |
| |
| static constexpr Device::Type | device_type_ |
| |
| static constexpr std::size_t | implementation_index_ |
| |