Searched refs:indicesShape (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | TopK_V2.cpp | 68 const Shape& valuesShape, void* indicesData, const Shape& indicesShape) { in eval() argument
73 reinterpret_cast<int32_t*>(indicesData), indicesShape); in eval()
78 reinterpret_cast<int32_t*>(indicesData), indicesShape); in eval()
83 reinterpret_cast<int32_t*>(indicesData), indicesShape); in eval()
88 reinterpret_cast<int32_t*>(indicesData), indicesShape); in eval()
|
| D | TopK_V2.h | 29 const Shape& valuesShape, void* indicesData, const Shape& indicesShape);
|
| D | Gather.cpp | 42 const int32_t* indicesData, const Shape& indicesShape, T* outputData) { in eval() argument
47 const auto indicesCount = getNumberOfElements(indicesShape); in eval()
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1721 Shape indicesShape = indices.shape(); in executeOperation() local
1723 success = topk_v2::prepare(input.shape(), k, &valuesShape, &indicesShape) && in executeOperation()
1725 setInfoAndAllocateIfNeeded(&indices, indicesShape, &result) && in executeOperation()
1727 indices.buffer, indicesShape); in executeOperation()
|