Searched refs:index_select_shape (Results 1 – 1 of 1) sorted by relevance
103 llvm::SmallVector<int64_t, 4> index_select_shape = in matchAndRewrite() local107 index_select_shape.push_back(dim); in matchAndRewrite()119 RankedTensorType::get(index_select_shape, operand_ty.getElementType()), in matchAndRewrite()