Searched refs:bShape (Results 1 – 6 of 6) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | PRelu.cpp | 46 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in eval() argument
49 IndexedShapeWrapper bShapeIndexed(bShape); in eval()
69 bool evalQuant8(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalQuant8() argument
72 const int32_t alpha_offset = -bShape.offset; in evalQuant8()
74 const double input_product_scale = aShape.scale * bShape.scale; in evalQuant8()
97 aData, aShape, bData, bShape, outputData, outputShape); in evalQuant8()
|
| 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()
|
| D | Comparisons.cpp | 35 const Shape& aShape, const DataType* bData, const Shape& bShape, bool8* outputData, in compute() argument
38 IndexedShapeWrapper bShapeIndexed(bShape); in compute()
53 const float realB = (bData[bFlatIndex] - bShape.offset) * bShape.scale; in compute()
|
| D | LogicalAndOr.cpp | 34 const bool8* bData, const Shape& bShape, bool8* outputData, const Shape& outputShape) { in compute() argument
36 IndexedShapeWrapper bShapeIndexed(bShape); in compute()
|
| D | Select.cpp | 32 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in compute() argument
43 b = requantize<T>(b, bShape, outputShape); in compute()
|
| D | Broadcast.cpp | 212 const Shape& bShape, int32_t activation, int32_t* outputData, in executeInt32() argument
216 IndexedShapeWrapper bShapeIndexed(bShape); in executeInt32()
|