Searched refs:aShape (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | PRelu.cpp | 41 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in eval() argument
43 IndexedShapeWrapper aShapeIndexed(aShape); in eval()
63 bool evalQuant8(const uint8_t* aData, const Shape& aShape, const uint8_t* bData, in evalQuant8() argument
65 const int32_t input_offset = -aShape.offset; in evalQuant8()
68 const double input_product_scale = aShape.scale * bShape.scale; in evalQuant8()
69 const double real_multiplier_pos = aShape.scale / outputShape.scale; in evalQuant8()
92 aData, aShape, bData, bShape, outputData, outputShape); in evalQuant8()
|
| D | MaximumMinimum.cpp | 31 bool evalGeneric(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalGeneric() argument
33 IndexedShapeWrapper aShapeIndexed(aShape); in evalGeneric()
56 bool evalQuant8(const uint8_t* aData, const Shape& aShape, const uint8_t* bData, in evalQuant8() argument
59 IndexedShapeWrapper aShapeIndexed(aShape); in evalQuant8()
73 uint8_t aValue = requantize(aData[aFlatIndex], aShape, outputShape); in evalQuant8()
|
| D | Select.cpp | 40 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in compute() argument
48 if (aShape.type == OperandType::TENSOR_QUANT8_ASYMM) { in compute()
49 a = requantize(a, aShape, outputShape); in compute()
|
| D | Comparisons.cpp | 39 const Shape& aShape, const DataType* bData, const Shape& bShape, bool8* outputData, in compute() argument
41 IndexedShapeWrapper aShapeIndexed(aShape); in compute()
54 if (aShape.type == OperandType::TENSOR_QUANT8_ASYMM) { in compute()
55 const float realA = (aData[aFlatIndex] - aShape.offset) * aShape.scale; in compute()
|
| D | LogicalAndOr.cpp | 37 bool compute(const std::function<bool(bool, bool)>& func, const bool8* aData, const Shape& aShape, in compute() argument
39 IndexedShapeWrapper aShapeIndexed(aShape); in compute()
|