ONE - On-device Neural Engine
|
#include <LogSoftmax.h>
Data Fields | |
float | beta |
int | axis |
Definition at line 35 of file LogSoftmax.h.
int onert::ir::operation::LogSoftmax::Param::axis |
Definition at line 38 of file LogSoftmax.h.
Referenced by metrics.categorical_accuracy.CategoricalAccuracy::update_state(), and onert::backend::cpu::KernelGenerator::visit().
float onert::ir::operation::LogSoftmax::Param::beta |
Definition at line 37 of file LogSoftmax.h.
Referenced by onert::backend::cpu::KernelGenerator::visit().