Searched refs:concat_result_shape (Results 1 – 1 of 1) sorted by relevance
604 SmallVector<int64_t, 4> concat_result_shape; in convertConcatV2Op() local622 concat_result_shape.push_back(2); in convertConcatV2Op()629 concat_result_shape.push_back(lhs_tensor_shape[i]); in convertConcatV2Op()631 concat_result_shape[axis] = lhs_tensor_shape[axis] + rhs_tensor_shape[axis]; in convertConcatV2Op()635 ArrayRef<int64_t>(concat_result_shape), result_type.getElementType()); in convertConcatV2Op()698 concat_result_shape[axis] = concat_result_shape[axis] + 1; in convertConcatV2Op()700 concat_result_shape[axis] = in convertConcatV2Op()701 concat_result_shape[axis] + rhs_tensor_shape[axis]; in convertConcatV2Op()703 concat_type = RankedTensorType::get(ArrayRef<int64_t>(concat_result_shape), in convertConcatV2Op()