Searched refs:isArgMin (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | ArgMinMax.cpp | 33 static void argMinMaxImpl(const In* inputData, const Shape& inputShape, int32_t axis, bool isArgMin, in argMinMaxImpl() argument
45 if ((isArgMin && value < minMaxValue) || (!isArgMin && value > minMaxValue)) { in argMinMaxImpl()
55 bool argMinMaxGeneric(const uint8_t* inputData, const Shape& inputShape, int32 axis, bool isArgMin, in argMinMaxGeneric() argument
63 argMinMaxImpl(reinterpret_cast<const dataType*>(inputData), inputShape, axis, isArgMin, \ in argMinMaxGeneric()
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 122 bool isArgMin, uint8_t* outputData, const Shape& outputShape);
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1403 const bool isArgMin = operation.type == OperationType::ARGMIN; in executeOperation() local
1406 argMinMaxGeneric(input.buffer, input.shape(), axis, isArgMin, output.buffer, in executeOperation()
|