InfiniOps
Operator Library for Accelerators
Loading...
Searching...
No Matches
infini::ops::NllLoss2dBackward Member List

This is the complete list of members for infini::ops::NllLoss2dBackward, including all inherited members.

active_implementation_indices(Device::Type dev_type)infini::ops::Operator< NllLoss2dBackward >inlinestatic
active_implementation_indices(Device::Type dev_type)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Call(const Handle &handle, const Config &config, const Args &... args)infini::ops::Operator< NllLoss2dBackward >static
Call(const Tensor tensor, const Args &... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Call(const TensorLike &tensor, const Args &... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Call(const Handle &handle, const Config &config, const Args &... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Call(const Tensor tensor, const Args &... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Call(const TensorLike &tensor, const Args &... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
clear_cache()infini::ops::Operator< NllLoss2dBackward >inlinestatic
clear_cache()infini::ops::Operator< NllLoss2dBackward >inlinestatic
config_ptr_infini::ops::OperatorBaseprotected
device_index_infini::ops::NllLoss2dBackwardprotected
device_type_infini::ops::Operator< NllLoss2dBackward >protectedstatic
grad_input_shape_infini::ops::NllLoss2dBackwardprotected
grad_input_strides_infini::ops::NllLoss2dBackwardprotected
grad_input_type_infini::ops::NllLoss2dBackwardprotected
grad_output_shape_infini::ops::NllLoss2dBackwardprotected
grad_output_strides_infini::ops::NllLoss2dBackwardprotected
grad_output_type_infini::ops::NllLoss2dBackwardprotected
handle_ptr_infini::ops::OperatorBaseprotected
has_weight_infini::ops::NllLoss2dBackwardprotected
ignore_index_infini::ops::NllLoss2dBackwardprotected
implementation_index_infini::ops::Operator< NllLoss2dBackward >protectedstatic
input_shape_infini::ops::NllLoss2dBackwardprotected
input_strides_infini::ops::NllLoss2dBackwardprotected
input_type_infini::ops::NllLoss2dBackwardprotected
Make(const Config &config, const Tensor tensor, Args &&... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Make(const Tensor tensor, Args &&... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Make(const std::vector< Tensor > tensors, Args &&... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Make(const Config &config, const Tensor tensor, Args &&... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Make(const Tensor tensor, Args &&... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Make(const Config &config, const std::vector< Tensor > tensors, Args &&... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
Make(const std::vector< Tensor > tensors, Args &&... args)infini::ops::Operator< NllLoss2dBackward >inlinestatic
NllLoss2dBackward(const Tensor grad_output, const Tensor input, const Tensor target, const std::optional< Tensor > weight, const Tensor total_weight, const int64_t reduction, const int64_t ignore_index, Tensor grad_input)infini::ops::NllLoss2dBackwardinline
operator()(const Tensor grad_output, const Tensor input, const Tensor target, const std::optional< Tensor > weight, const Tensor total_weight, const int64_t reduction, const int64_t ignore_index, Tensor grad_input) const =0infini::ops::NllLoss2dBackwardpure virtual
Operator< NllLoss2dBackward >::operator()(const Handle &handle, const Args &... args)infini::ops::Operator< NllLoss2dBackward >inline
Operator< NllLoss2dBackward >::operator()(const Args &... args) constinfini::ops::Operator< NllLoss2dBackward >inline
reduction_infini::ops::NllLoss2dBackwardprotected
set_config(const Config &config)infini::ops::OperatorBaseinline
set_config(const Config &config)infini::ops::OperatorBaseinline
set_handle(const Handle &handle)infini::ops::OperatorBaseinline
set_handle(const Handle &handle)infini::ops::OperatorBaseinline
set_stream(void *stream)infini::ops::OperatorBaseinline
set_stream(void *stream)infini::ops::OperatorBaseinline
set_workspace(void *workspace)infini::ops::OperatorBaseinline
set_workspace(void *workspace)infini::ops::OperatorBaseinline
set_workspace_size_in_bytes(std::size_t workspace_size_in_bytes)infini::ops::OperatorBaseinline
set_workspace_size_in_bytes(std::size_t workspace_size_in_bytes)infini::ops::OperatorBaseinline
stream_infini::ops::OperatorBaseprotected
target_shape_infini::ops::NllLoss2dBackwardprotected
target_strides_infini::ops::NllLoss2dBackwardprotected
target_type_infini::ops::NllLoss2dBackwardprotected
total_weight_shape_infini::ops::NllLoss2dBackwardprotected
total_weight_strides_infini::ops::NllLoss2dBackwardprotected
total_weight_type_infini::ops::NllLoss2dBackwardprotected
weight_shape_infini::ops::NllLoss2dBackwardprotected
weight_strides_infini::ops::NllLoss2dBackwardprotected
weight_type_infini::ops::NllLoss2dBackwardprotected
workspace_infini::ops::OperatorBaseprotected
workspace_size_in_bytes() constinfini::ops::OperatorBaseinlinevirtual
workspace_size_in_bytes() constinfini::ops::OperatorBaseinlinevirtual
workspace_size_in_bytes_infini::ops::OperatorBaseprotected
~OperatorBase()=defaultinfini::ops::OperatorBasevirtual
~OperatorBase()=defaultinfini::ops::OperatorBasevirtual