Home
last modified time | relevance | path

Searched refs:SameShape (Results 1 – 5 of 5) sorted by relevance

/frameworks/ml/nn/common/operations/
DPow.cpp61 if (SameShape(baseShape, exponentShape)) { in prepare()
DBroadcast.cpp87 bool needBroadcast = !SameShape(shape1, shape2); in addFloat32()
120 bool needBroadcast = !SameShape(shape1, shape2); in addQuant8()
185 bool needBroadcast = !SameShape(shape1, shape2); in mulFloat32()
327 bool needBroadcast = !SameShape(shape1, shape2); in divFloat32()
DSelect.cpp97 NN_RET_CHECK(SameShape(input1, input2)); in prepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h136 bool SameShape(const Shape& in1, const Shape& in2);
/frameworks/ml/nn/common/
DOperationsUtils.cpp68 bool SameShape(const Shape& in1, const Shape& in2) { in SameShape() function