Searched refs:updates_type (Results 1 – 4 of 4) sorted by relevance
292 auto updates_type = updates.getType().cast<RankedTensorType>(); in ScatterAccumulateElements() local294 if (maybe_contiguous_start == 0 && buffer_type == updates_type) { in ScatterAccumulateElements()305 for (int64_t i = 0; i < updates_type.getDimSize(0); ++i) { in ScatterAccumulateElements()
1099 ShapedType updates_type = updates.getType().cast<ShapedType>(); in matchAndRewrite() local1103 !updates_type.hasStaticShape()) { in matchAndRewrite()
1034 auto updates_type = updates.getType().cast<ShapedType>(); in Verify() local1037 if (!indices_type.hasStaticShape() || !updates_type.hasStaticShape()) { in Verify()1050 if (indices_type.getDimSize(i) != updates_type.getDimSize(i)) { in Verify()1054 << ") == " << updates_type.getDimSize(i); in Verify()1062 auto output_rank = outermost_dim + updates_type.getRank() - outer_dims; in Verify()1091 if (value != updates_type.getDimSize(corresponding_dim)) { in Verify()1094 << ") == " << updates_type.getDimSize(corresponding_dim) in Verify()
5179 auto updates_type = updates.getType().cast<ShapedType>(); in matchAndRewrite() local5198 split_updates_shape.append(updates_type.getShape().begin(), in matchAndRewrite()5199 updates_type.getShape().end()); in matchAndRewrite()5203 updates_type.getShape().front(), in matchAndRewrite()5205 updates_type.getElementType())); in matchAndRewrite()5215 SmallVector<int64_t, 6> starts(updates_type.getRank(), 0); in matchAndRewrite()5216 SmallVector<int64_t, 6> strides(updates_type.getRank(), 1); in matchAndRewrite()5217 SmallVector<int64_t, 6> limits(updates_type.getShape().begin(), in matchAndRewrite()5218 updates_type.getShape().end()); in matchAndRewrite()