|
ONE - On-device Neural Engine
|
#include <KernelParams.h>

Data Fields | |
| int32_t | axis |
| int32_t | batch_dims |
Definition at line 103 of file KernelParams.h.
| int32_t luci_interpreter::GatherParams::axis |
Definition at line 105 of file KernelParams.h.
Referenced by onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::__init__(), luci_interpreter::build_kernel_CircleGather(), luci_interpreter::kernels::Gather::configure(), and onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::update_state().
| int32_t luci_interpreter::GatherParams::batch_dims |
Definition at line 106 of file KernelParams.h.
Referenced by luci_interpreter::kernels::Gather::configure().