Searched refs:baseShape (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Pow.cpp | 34 bool evalGeneric(const T* baseData, const Shape& baseShape, const T* exponentData, in evalGeneric() argument
36 IndexedShapeWrapper baseShapeIndexed(baseShape); in evalGeneric()
61 bool prepare(const Shape& baseShape, const Shape& exponentShape, Shape* output) { in prepare() argument
62 NN_OPS_CHECK(baseShape.type == exponentShape.type); in prepare()
63 if (SameShape(baseShape, exponentShape)) { in prepare()
64 return SetShape(baseShape, output); in prepare()
66 return calculateBroadcastedShape(baseShape, exponentShape, output); in prepare()
69 bool eval(const void* baseData, const Shape& baseShape, const void* exponentData, in eval() argument
71 switch (baseShape.type) { in eval()
73 return evalGeneric(reinterpret_cast<const _Float16*>(baseData), baseShape, in eval()
[all …]
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | Pow.h | 28 bool eval(const void* baseData, const Shape& baseShape, const void* exponentData,
|