Searched refs:split_updates_shape (Results 1 – 1 of 1) sorted by relevance
5197 SmallVector<int64_t, 4> split_updates_shape; in matchAndRewrite() local5198 split_updates_shape.append(updates_type.getShape().begin(), in matchAndRewrite()5200 split_updates_shape.front() = 1; in matchAndRewrite()5204 RankedTensorType::get(split_updates_shape, in matchAndRewrite()