Searched refs:output_shape_vals (Results 1 – 2 of 2) sorted by relevance
153 SmallVector<int64_t, 8> output_shape_vals(result_type.getShape().begin(), in convertPackOp() local155 if (output_shape_vals.size() != (input_tensor_rank + 1)) { in convertPackOp()160 if (output_shape_vals[axis] != inputs.size()) { in convertPackOp()187 reshape_output_shape.push_back(output_shape_vals[axis]); in convertPackOp()190 reshape_output_shape.push_back(output_shape_vals[d]); in convertPackOp()196 reshape_output_shape.assign(output_shape_vals.begin(), in convertPackOp()197 output_shape_vals.end()); in convertPackOp()
998 SmallVector<Value, 4> output_shape_vals{output_batch}; in matchAndRewrite() local1000 output_shape_vals.push_back(outer_shape_vals[i]); in matchAndRewrite()1003 output_shape_vals.push_back(padded_shape_splits[i]); in matchAndRewrite()1006 rewriter, loc, rewriter.getIntegerType(64), output_shape_vals); in matchAndRewrite()