Searched refs:baseShape (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Pow.cpp | 35 bool evalGeneric(const T* baseData, const Shape& baseShape, const T* exponentData, in evalGeneric() argument
37 IndexedShapeWrapper baseShapeIndexed(baseShape); in evalGeneric()
62 bool prepare(const Shape& baseShape, const Shape& exponentShape, Shape* output) { in prepare() argument
63 NN_OPS_CHECK(baseShape.type == exponentShape.type); in prepare()
64 if (SameShape(baseShape, exponentShape)) { in prepare()
65 return SetShape(baseShape, output); in prepare()
67 return calculateBroadcastedShape(baseShape, exponentShape, output); in prepare()
70 bool eval(const void* baseData, const Shape& baseShape, const void* exponentData, in eval() argument
72 switch (baseShape.type) { in eval()
74 return evalGeneric(reinterpret_cast<const _Float16*>(baseData), baseShape, in eval()
[all …]
|
| D | Pow.h | 28 bool eval(const void* baseData, const Shape& baseShape, const void* exponentData,
|