Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DTopK_V2.cpp35 int32_t* indicesData) { in evalGeneric() argument
40 int32_t* curOutputIndex = indicesData; in evalGeneric()
DGather.cpp32 const int32_t* indicesData, const Shape& indicesShape, T* outputData) { in eval() argument
40 const auto inputIndex = static_cast<uint32_t>(indicesData[outputIndex]); in eval()