Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DTopK_V2.cpp42 int32_t* indicesData) { in evalGeneric() argument
47 int32_t* curOutputIndex = indicesData; in evalGeneric()
H A DGather.cpp41 const int32_t* indicesData, const Shape& indicesShape, T* outputData) { in eval() argument
49 const auto inputIndex = static_cast<uint32_t>(indicesData[outputIndex]); in eval()