Searched refs:bShape (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
40 IndexedShapeWrapper bShapeIndexed(bShape); in evalGeneric()
63 bool evalQuant8(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalQuant8() argument
66 IndexedShapeWrapper bShapeIndexed(bShape); in evalQuant8()
80 T bValue = requantize<T>(bData[bFlatIndex], bShape, outputShape); in evalQuant8()
|
| 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()
|
| D | LogicalAndOr.cpp | 40 const bool8* bData, const Shape& bShape, bool8* outputData, const Shape& outputShape) { in compute() argument
42 IndexedShapeWrapper bShapeIndexed(bShape); in compute()
|
| D | Comparisons.cpp | 44 const Shape& aShape, const DataType* bData, const Shape& bShape, bool8* outputData, in compute() argument
47 IndexedShapeWrapper bShapeIndexed(bShape); in compute()
62 const float realB = (bData[bFlatIndex] - bShape.offset) * bShape.scale; in compute()
|
| D | Select.cpp | 42 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in compute() argument
53 b = requantize<T>(b, bShape, outputShape); in compute()
|
| D | Broadcast.cpp | 209 const Shape& bShape, int32_t activation, int32_t* outputData, in executeInt32() argument
213 IndexedShapeWrapper bShapeIndexed(bShape); in executeInt32()
|