Searched refs:valuesShape (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | TopK_V2.cpp | 68 const Shape& valuesShape, void* indicesData, const Shape& indicesShape) { in eval() argument
72 reinterpret_cast<_Float16*>(valuesData), valuesShape, in eval()
77 reinterpret_cast<float*>(valuesData), valuesShape, in eval()
82 reinterpret_cast<int32_t*>(valuesData), valuesShape, in eval()
87 reinterpret_cast<uint8_t*>(valuesData), valuesShape, in eval()
|
| D | TopK_V2.h | 29 const Shape& valuesShape, void* indicesData, const Shape& indicesShape);
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1719 Shape valuesShape = values.shape(); in executeOperation() local
1723 success = topk_v2::prepare(input.shape(), k, &valuesShape, &indicesShape) && in executeOperation()
1724 setInfoAndAllocateIfNeeded(&values, valuesShape, &result) && in executeOperation()
1726 topk_v2::eval(input.buffer, input.shape(), k, values.buffer, valuesShape, in executeOperation()
|