Searched refs:isArgMin (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | ArgMinMax.cpp | 31 static void argMinMaxImpl(const In* inputData, const Shape& inputShape, int32_t axis, bool isArgMin, in argMinMaxImpl() argument
43 if ((isArgMin && value < minMaxValue) || (!isArgMin && value > minMaxValue)) { in argMinMaxImpl()
53 bool argMinMaxGeneric(const uint8_t* inputData, const Shape& inputShape, int32 axis, bool isArgMin, in argMinMaxGeneric() argument
61 argMinMaxImpl(reinterpret_cast<const dataType*>(inputData), inputShape, axis, isArgMin, \ in argMinMaxGeneric()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | Operations.h | 109 bool isArgMin, uint8_t* outputData, const Shape& outputShape);
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1348 const bool isArgMin = operation.type == OperationType::ARGMIN; in executeOperation() local
1351 argMinMaxGeneric(input.buffer, input.shape(), axis, isArgMin, output.buffer, in executeOperation()
|