Searched refs:reshape_output_shape (Results 1 – 1 of 1) sorted by relevance
232 SmallVector<int64_t> reshape_output_shape; in convertPackOp() local242 reshape_output_shape.push_back(output_shape_vals[axis]); in convertPackOp()245 reshape_output_shape.push_back(output_shape_vals[d]); in convertPackOp()251 reshape_output_shape.assign(output_shape_vals.begin(), in convertPackOp()255 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_output_shape); in convertPackOp()285 RankedTensorType::get(reshape_output_shape, result_type.getElementType()); in convertPackOp()