Searched refs:bShape (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | MaximumMinimum.cpp | 31 bool evalGeneric(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalGeneric() argument
34 IndexedShapeWrapper bShapeIndexed(bShape); in evalGeneric()
57 const Shape& bShape, bool isMinimum, uint8_t* outputData, in evalQuant8() argument
60 IndexedShapeWrapper bShapeIndexed(bShape); in evalQuant8()
74 uint8_t bValue = requantize(bData[bFlatIndex], bShape, outputShape); in evalQuant8()
|
| D | PRelu.cpp | 41 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in eval() argument
44 IndexedShapeWrapper bShapeIndexed(bShape); in eval()
64 const Shape& bShape, uint8_t* outputData, const Shape& outputShape) { in evalQuant8() argument
66 const int32_t alpha_offset = -bShape.offset; in evalQuant8()
68 const double input_product_scale = aShape.scale * bShape.scale; in evalQuant8()
92 aData, aShape, bData, bShape, outputData, outputShape); in evalQuant8()
|
| D | Comparisons.cpp | 39 const Shape& aShape, const DataType* bData, const Shape& bShape, bool8* outputData, in compute() argument
42 IndexedShapeWrapper bShapeIndexed(bShape); in compute()
56 const float realB = (bData[bFlatIndex] - bShape.offset) * bShape.scale; in compute()
|
| D | LogicalAndOr.cpp | 38 const bool8* bData, const Shape& bShape, bool8* outputData, const Shape& outputShape) { in compute() argument
40 IndexedShapeWrapper bShapeIndexed(bShape); in compute()
|
| D | Select.cpp | 40 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in compute() argument
50 b = requantize(b, bShape, outputShape); in compute()
|