- Generated by
1.9.8
|
InfiniOps
Operator Library for Accelerators
|
This is the complete list of members for infini::ops::NllLoss, including all inherited members.
| active_implementation_indices(Device::Type dev_type) | infini::ops::Operator< NllLoss > | inlinestatic |
| active_implementation_indices(Device::Type dev_type) | infini::ops::Operator< NllLoss > | inlinestatic |
| Call(const Handle &handle, const Config &config, const Args &... args) | infini::ops::Operator< NllLoss > | static |
| Call(const Tensor tensor, const Args &... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| Call(const TensorLike &tensor, const Args &... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| Call(const Handle &handle, const Config &config, const Args &... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| Call(const Tensor tensor, const Args &... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| Call(const TensorLike &tensor, const Args &... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| clear_cache() | infini::ops::Operator< NllLoss > | inlinestatic |
| clear_cache() | infini::ops::Operator< NllLoss > | inlinestatic |
| config_ptr_ | infini::ops::OperatorBase | protected |
| device_index_ | infini::ops::NllLoss | protected |
| device_type_ | infini::ops::Operator< NllLoss > | protectedstatic |
| handle_ptr_ | infini::ops::OperatorBase | protected |
| has_weight_ | infini::ops::NllLoss | protected |
| ignore_index_ | infini::ops::NllLoss | protected |
| implementation_index_ | infini::ops::Operator< NllLoss > | protectedstatic |
| input_shape_ | infini::ops::NllLoss | protected |
| input_strides_ | infini::ops::NllLoss | protected |
| input_type_ | infini::ops::NllLoss | protected |
| Make(const Config &config, const Tensor tensor, Args &&... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| Make(const Tensor tensor, Args &&... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| Make(const std::vector< Tensor > tensors, Args &&... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| Make(const Config &config, const Tensor tensor, Args &&... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| Make(const Tensor tensor, Args &&... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| Make(const std::vector< Tensor > tensors, Args &&... args) | infini::ops::Operator< NllLoss > | inlinestatic |
| NllLoss(const Tensor input, const Tensor target, const std::optional< Tensor > weight, const std::optional< bool > size_average, const int64_t ignore_index, const std::optional< bool > reduce, const std::string reduction, Tensor out) | infini::ops::NllLoss | inline |
| NllLoss(const Tensor input, const Tensor target, const std::optional< Tensor > weight, const int64_t reduction, const int64_t ignore_index, Tensor out) | infini::ops::NllLoss | inline |
| operator()(const Tensor input, const Tensor target, const std::optional< Tensor > weight, const std::optional< bool > size_average, const int64_t ignore_index, const std::optional< bool > reduce, const std::string reduction, Tensor out) const | infini::ops::NllLoss | inline |
| operator()(const Tensor input, const Tensor target, const std::optional< Tensor > weight, const int64_t reduction, const int64_t ignore_index, Tensor out) const =0 | infini::ops::NllLoss | pure virtual |
| Operator< NllLoss >::operator()(const Handle &handle, const Args &... args) | infini::ops::Operator< NllLoss > | inline |
| Operator< NllLoss >::operator()(const Args &... args) const | infini::ops::Operator< NllLoss > | inline |
| out_shape_ | infini::ops::NllLoss | protected |
| out_strides_ | infini::ops::NllLoss | protected |
| out_type_ | infini::ops::NllLoss | protected |
| reduction_ | infini::ops::NllLoss | protected |
| set_config(const Config &config) | infini::ops::OperatorBase | inline |
| set_config(const Config &config) | infini::ops::OperatorBase | inline |
| set_handle(const Handle &handle) | infini::ops::OperatorBase | inline |
| set_handle(const Handle &handle) | infini::ops::OperatorBase | inline |
| set_stream(void *stream) | infini::ops::OperatorBase | inline |
| set_stream(void *stream) | infini::ops::OperatorBase | inline |
| set_workspace(void *workspace) | infini::ops::OperatorBase | inline |
| set_workspace(void *workspace) | infini::ops::OperatorBase | inline |
| set_workspace_size_in_bytes(std::size_t workspace_size_in_bytes) | infini::ops::OperatorBase | inline |
| set_workspace_size_in_bytes(std::size_t workspace_size_in_bytes) | infini::ops::OperatorBase | inline |
| stream_ | infini::ops::OperatorBase | protected |
| target_shape_ | infini::ops::NllLoss | protected |
| target_strides_ | infini::ops::NllLoss | protected |
| target_type_ | infini::ops::NllLoss | protected |
| weight_shape_ | infini::ops::NllLoss | protected |
| weight_strides_ | infini::ops::NllLoss | protected |
| weight_type_ | infini::ops::NllLoss | protected |
| workspace_ | infini::ops::OperatorBase | protected |
| workspace_size_in_bytes() const | infini::ops::OperatorBase | inlinevirtual |
| workspace_size_in_bytes() const | infini::ops::OperatorBase | inlinevirtual |
| workspace_size_in_bytes_ | infini::ops::OperatorBase | protected |
| ~OperatorBase()=default | infini::ops::OperatorBase | virtual |
| ~OperatorBase()=default | infini::ops::OperatorBase | virtual |
1.9.8