Searched refs:aData (Results 1 – 6 of 6) sorted by relevance
/frameworks/ml/nn/common/operations/ |
D | MaximumMinimum.cpp | 37 bool evalGeneric(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalGeneric() argument 53 outputData[outputFlatIndex] = isMinimum ? std::min(aData[aFlatIndex], bData[bFlatIndex]) in evalGeneric() 54 : std::max(aData[aFlatIndex], bData[bFlatIndex]); in evalGeneric() 63 bool evalQuant8(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalQuant8() argument 79 T aValue = requantize<T>(aData[aFlatIndex], aShape, outputShape); in evalQuant8()
|
D | PRelu.cpp | 46 inline bool eval(const std::function<T(const T&, const T&)>& func, const T* aData, in eval() argument 62 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in eval() 70 bool evalQuant8(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalQuant8() argument 98 aData, aShape, bData, bShape, outputData, outputShape); in evalQuant8()
|
D | LogicalAndOr.cpp | 39 bool compute(const std::function<bool(bool, bool)>& func, const bool8* aData, const Shape& aShape, in compute() argument 54 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in compute()
|
D | Select.cpp | 41 bool compute(const bool8* conditionData, const Shape& conditionShape, const T* aData, in compute() argument 48 T a = aData[i]; in compute()
|
D | Comparisons.cpp | 43 bool compute(const std::function<bool(ComparisonType, ComparisonType)>& func, const DataType* aData, in compute() argument 61 const float realA = (aData[aFlatIndex] - aShape.offset) * aShape.scale; in compute() 65 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in compute()
|
D | Broadcast.cpp | 208 bool executeInt32(const int32_t* aData, const Shape& aShape, const int32_t* bData, in executeInt32() argument 225 outputData[outputFlatIndex] = func(aData[aFlatIndex], bData[bFlatIndex]); in executeInt32()
|