ONE - On-device Neural Engine
|
#include <KernelParams.h>
Data Fields | |
int32_t | axis |
Definition at line 155 of file KernelParams.h.
int32_t luci_interpreter::OneHotParams::axis |
Definition at line 157 of file KernelParams.h.
Referenced by package.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::__init__(), luci_interpreter::build_kernel_CircleOneHot(), luci_interpreter::kernels::OneHot::configure(), luci_interpreter::kernels::OneHot::execute(), and package.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::update_state().