Home
last modified time | relevance | path

Searched refs:kOutputIndicesTensor (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DTopK_V2.cpp64 context->getOutputBuffer<int32_t>(kOutputIndicesTensor)); in executeTyped()
84 context->setOutputShape(kOutputIndicesTensor, outputIndicesShape); in prepare()
/packages/modules/NeuralNetworks/common/types/operations/include/
DTopK_V2.h30 constexpr uint32_t kOutputIndicesTensor = 1; variable