Searched refs:bShape (Results 1 – 6 of 6) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | MaximumMinimum.cpp | 35 bool evalGeneric(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalGeneric() argument
38 IndexedShapeWrapper bShapeIndexed(bShape); in evalGeneric()
61 bool evalQuant8(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalQuant8() argument
64 IndexedShapeWrapper bShapeIndexed(bShape); in evalQuant8()
78 T bValue = requantize<T>(bData[bFlatIndex], bShape, outputShape); in evalQuant8()
|
| H A D | PRelu.cpp | 47 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in eval() argument
50 IndexedShapeWrapper bShapeIndexed(bShape); in eval()
70 bool evalQuant8(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalQuant8() argument
73 const int32_t alpha_offset = -bShape.offset; in evalQuant8()
75 const double input_product_scale = aShape.scale * bShape.scale; in evalQuant8()
98 aData, aShape, bData, bShape, outputData, outputShape); in evalQuant8()
|
| H A D | Comparisons.cpp | 41 const Shape& aShape, const DataType* bData, const Shape& bShape, bool8* outputData, in compute() argument
44 IndexedShapeWrapper bShapeIndexed(bShape); in compute()
59 const float realB = (bData[bFlatIndex] - bShape.offset) * bShape.scale; in compute()
|
| H A D | LogicalAndOr.cpp | 40 const bool8* bData, const Shape& bShape, bool8* outputData, const Shape& outputShape) { in compute() argument
42 IndexedShapeWrapper bShapeIndexed(bShape); in compute()
|
| H A D | Select.cpp | 39 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in compute() argument
50 b = requantize<T>(b, bShape, outputShape); in compute()
|
| H A D | Broadcast.cpp | 212 const Shape& bShape, int32_t activation, int32_t* outputData, in executeInt32() argument
216 IndexedShapeWrapper bShapeIndexed(bShape); in executeInt32()
|