Searched refs:concat_output_shape (Results 1 – 1 of 1) sorted by relevance
206 SmallVector<int64_t, 4> concat_output_shape; in convertPackOp() local208 concat_output_shape.push_back(1); in convertPackOp()212 concat_output_shape.push_back(input0_tensor_shape[i]); in convertPackOp()217 concat_output_shape[concat_axis] = orig_input_dim_on_axis * 2; in convertPackOp()219 ArrayRef<int64_t>(concat_output_shape), result_type.getElementType()); in convertPackOp()226 concat_output_shape[concat_axis] = orig_input_dim_on_axis * (i + 1); in convertPackOp()227 concat_type = RankedTensorType::get(ArrayRef<int64_t>(concat_output_shape), in convertPackOp()