24.02.1
|
Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator. More...
#include <SoftmaxAttributes.h>
Public Member Functions | |
SoftmaxAttributes & | beta (float beta) |
Set beta. More... | |
float | beta () const |
Get beta. More... | |
SoftmaxAttributes & | is_log_softmax (bool is_log) |
Set is_log. More... | |
bool | is_log_softmax () const |
Get is_log. More... | |
SoftmaxAttributes & | axis (int axis) |
Set axis. More... | |
int | axis () const |
Get axis. More... | |
Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator.
Softmax attributes
Definition at line 37 of file SoftmaxAttributes.h.
int axis | ( | ) | const |
Get axis.
Definition at line 61 of file SoftmaxAttributes.cpp.
Referenced by SoftmaxAttributes::axis().
SoftmaxAttributes & axis | ( | int | axis | ) |
Set axis.
Definition at line 55 of file SoftmaxAttributes.cpp.
References SoftmaxAttributes::axis().
Referenced by arm_compute::operator<<(), and GpuSoftmax::validate_op().
float beta | ( | ) | const |
Get beta.
Definition at line 39 of file SoftmaxAttributes.cpp.
Referenced by SoftmaxAttributes::beta().
SoftmaxAttributes & beta | ( | float | beta | ) |
Set beta.
Definition at line 33 of file SoftmaxAttributes.cpp.
References SoftmaxAttributes::beta().
Referenced by ClTemplateLogits1DMaxShiftExpSum::get_component_code(), ClTemplateLogits1DMaxShiftExpSum::get_tag_lut(), and arm_compute::operator<<().
bool is_log_softmax | ( | ) | const |
Get is_log.
Definition at line 50 of file SoftmaxAttributes.cpp.
Referenced by SoftmaxAttributes::is_log_softmax().
SoftmaxAttributes & is_log_softmax | ( | bool | is_log | ) |
Set is_log.
Definition at line 44 of file SoftmaxAttributes.cpp.
References SoftmaxAttributes::is_log_softmax().
Referenced by ClTemplateLogits1DNorm::get_component_code(), ClTemplateLogits1DMaxShiftExpSum::get_component_code(), arm_compute::operator<<(), and ClComponentLogits1DNorm::validate().