Searched refs:isArgMin (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | ArgMinMax.cpp | 31 int32_t axis, bool isArgMin, in argMinMaxImpl() argument
44 if ((isArgMin && value < minMaxValue) || in argMinMaxImpl()
45 (!isArgMin && value > minMaxValue)) { in argMinMaxImpl()
56 int32 axis, bool isArgMin, in argMinMaxGeneric() argument
68 isArgMin, \ in argMinMaxGeneric()
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 123 bool isArgMin, uint8_t* outputData, const Shape& outputShape);
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1441 const bool isArgMin = operation.type == OperationType::ARGMIN; in executeOperation() local
1444 argMinMaxGeneric(input.buffer, input.shape(), axis, isArgMin, output.buffer, in executeOperation()
|