Home
last modified time | relevance | path

Searched refs:SetShape (Results 1 – 8 of 8) sorted by relevance

/frameworks/ml/nn/common/operations/
DElementwise.cpp137 NN_RET_CHECK(SetShape(input, &output)); in prepare()
145 NN_RET_CHECK(SetShape(input, &output)); in prepareFloor()
DLogicalNot.cpp61 NN_RET_CHECK(SetShape(input, &output)); in prepare()
DPow.cpp65 return SetShape(baseShape, output); in prepare()
DNeg.cpp68 NN_RET_CHECK(SetShape(input, &output)); in prepare()
DSelect.cpp105 NN_RET_CHECK(SetShape(input1, &output)); in prepare()
DStridedSlice.cpp200 NN_RET_CHECK(SetShape(inputShape, &outputShape)); in prepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h143 bool SetShape(const Shape& in, Shape* out);
/frameworks/ml/nn/common/
DOperationsUtils.cpp128 bool SetShape(const Shape& in, Shape* out) { in SetShape() function