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