Home
last modified time | relevance | path

Searched refs:kInputAxis (Results 1 – 6 of 6) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DChannelShuffle.cpp55 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
64 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
DGather.cpp55 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
73 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
DLogSoftmax.cpp70 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
/packages/modules/NeuralNetworks/common/types/operations/include/
DGather.h28 constexpr uint32_t kInputAxis = 1; variable
DChannelShuffle.h29 constexpr uint32_t kInputAxis = 2; variable
DLogSoftmax.h29 constexpr uint32_t kInputAxis = 2; variable