Searched refs:SetShape (Results 1 – 8 of 8) sorted by relevance
61 NN_RET_CHECK(SetShape(input, &output)); in prepare()
137 NN_RET_CHECK(SetShape(input, &output)); in prepare()145 NN_RET_CHECK(SetShape(input, &output)); in prepareFloor()
65 return SetShape(baseShape, output); in prepare()
68 NN_RET_CHECK(SetShape(input, &output)); in prepare()
105 NN_RET_CHECK(SetShape(input1, &output)); in prepare()
200 NN_RET_CHECK(SetShape(inputShape, &outputShape)); in prepare()
143 bool SetShape(const Shape& in, Shape* out);
128 bool SetShape(const Shape& in, Shape* out) { in SetShape() function