Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DChannelShuffle.cpp35 constexpr uint32_t kInputAxis = 2; variable
88 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
97 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
DGather.cpp32 constexpr uint32_t kInputAxis = 1; variable
87 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
105 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
DLogSoftmax.cpp37 constexpr uint32_t kInputAxis = 2; variable
100 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()