Searched refs:SetShape (Results 1 – 8 of 8) sorted by relevance
131 NN_RET_CHECK(SetShape(input, &output)); in prepare()139 NN_RET_CHECK(SetShape(input, &output)); in prepareFloor()
58 NN_RET_CHECK(SetShape(input, &output)); in prepare()
64 return SetShape(baseShape, output); in prepare()
65 NN_RET_CHECK(SetShape(input, &output)); in prepare()
102 NN_RET_CHECK(SetShape(input1, &output)); in prepare()
203 NN_RET_CHECK(SetShape(inputShape, &outputShape)); in prepare()
127 bool SetShape(const Shape& in, Shape* out);
126 bool SetShape(const Shape& in, Shape* out) { in SetShape() function