Lines Matching refs:baseShape
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()
79 return evalGeneric(reinterpret_cast<const float*>(baseData), baseShape, in eval()
84 LOG(ERROR) << "Unsupported data type: " << toString(baseShape.type); in eval()