Searched refs:update_dim (Results 1 – 2 of 2) sorted by relevance
2903 int64_t update_dim = in HandleScatter() local2909 int64_t update_dim_size = update_dim == -1 ? 1 : updates_dims[update_dim]; in HandleScatter()
852 update_dim = lhs_scatter_update->shape().rank() - 1; in HandleAdd() local858 absl::c_linear_search(lhs_dnums.update_window_dims(), update_dim)) { in HandleAdd()859 --update_dim; in HandleAdd()864 first_update_dim = update_dim--; in HandleAdd()872 update_concat_dimension = update_dim--; in HandleAdd()