Searched refs:aShape (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | PRelu.cpp | 47 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in eval() argument
49 IndexedShapeWrapper aShapeIndexed(aShape); in eval()
70 bool evalQuant8(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalQuant8() argument
72 const int32_t input_offset = -aShape.offset; in evalQuant8()
75 const double input_product_scale = aShape.scale * bShape.scale; in evalQuant8()
76 const double real_multiplier_pos = aShape.scale / outputShape.scale; in evalQuant8()
98 aData, aShape, bData, bShape, outputData, outputShape); in evalQuant8()
|
| D | MaximumMinimum.cpp | 37 bool evalGeneric(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalGeneric() argument
39 IndexedShapeWrapper aShapeIndexed(aShape); in evalGeneric()
63 bool evalQuant8(const T* aData, const Shape& aShape, const T* bData, const Shape& bShape, in evalQuant8() argument
65 IndexedShapeWrapper aShapeIndexed(aShape); in evalQuant8()
79 T aValue = requantize<T>(aData[aFlatIndex], aShape, outputShape); in evalQuant8()
|
| D | Comparisons.cpp | 44 const Shape& aShape, const DataType* bData, const Shape& bShape, bool8* outputData, in compute() argument
46 IndexedShapeWrapper aShapeIndexed(aShape); in compute()
59 if (aShape.type == OperandType::TENSOR_QUANT8_ASYMM || in compute()
60 aShape.type == OperandType::TENSOR_QUANT8_ASYMM_SIGNED) { in compute()
61 const float realA = (aData[aFlatIndex] - aShape.offset) * aShape.scale; in compute()
|
| D | LogicalAndOr.cpp | 39 bool compute(const std::function<bool(bool, bool)>& func, const bool8* aData, const Shape& aShape, in compute() argument
41 IndexedShapeWrapper aShapeIndexed(aShape); in compute()
|
| D | Select.cpp | 42 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in compute() argument
52 a = requantize<T>(a, aShape, outputShape); in compute()
|
| D | Broadcast.cpp | 208 bool executeInt32(const int32_t* aData, const Shape& aShape, const int32_t* bData, in executeInt32() argument
212 IndexedShapeWrapper aShapeIndexed(aShape); in executeInt32()
|