Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Evict支持scatter_update操作吗 #1007

Open
surpercode opened this issue Aug 15, 2024 · 1 comment
Open

Evict支持scatter_update操作吗 #1007

surpercode opened this issue Aug 15, 2024 · 1 comment

Comments

@surpercode
Copy link

Evict支持scatter_update操作吗
ev_opt = tf.EmbeddingVariableOption(init_option=init_opt, filter_option=filter_opt, evict_option=evict_opt)
emb_table = tf.get_embedding_variable("ev_emb_table",
embedding_dim=64,
partitioner=tf.fixed_size_partitioner(num_shards=10),
ev_option=ev_opt)
emb_table能像普通embedding_table一样支持scatter_update操作吗

@candyzone
Copy link
Collaborator

目前不支持,对于emb table 的写操作,目前都是对应的Optimizer的Apply算子实现的

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants