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.

Sign up for GitHub

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

[Embedding] Add GPU fused embedding ops. #64

Open
wants to merge 49 commits into
base: main
Choose a base branch
from
Open

[Embedding] Add GPU fused embedding ops. #64

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
49 commits
Select commit Hold shift + click to select a range
a566c1c
Update fused embedding modelzoo perf benchmark
RandyInterfish Jan 10, 2022
4567918
Update perf benchmark under docs
RandyInterfish Jan 10, 2022
222e123
Merge branch 'main' of https://github.com/nvzhou/DeepRec-public into …
RandyInterfish Jan 10, 2022
fe129e8
Add nvtx to fused embedding ops
RandyInterfish Jan 14, 2022
13906e4
refactor a little big
RandyInterfish Jan 20, 2022
95a08ff
Update: add unique sub op and use cu event to sync
RandyInterfish Jan 27, 2022
f2ca5ea
Minor change
RandyInterfish Feb 9, 2022
1cf4c57
minor update
RandyInterfish Feb 10, 2022
22e936e
minor update
RandyInterfish Feb 11, 2022
3fab0e8
Update: minor change
RandyInterfish Feb 14, 2022
d9765a5
temp update
RandyInterfish Feb 14, 2022
3b00778
kernel impl compile pass
RandyInterfish Feb 15, 2022
7589dc1
Update: pre-lookup ready. Unit tests all passed
RandyInterfish Feb 22, 2022
d65cec1
postlookup grad done except for unit test
RandyInterfish Mar 1, 2022
cca0b38
Unit test passed
RandyInterfish Mar 10, 2022
279a6de
python api works right
RandyInterfish Mar 10, 2022
6569c30
Update: try to optimize
RandyInterfish Mar 15, 2022
6c29295
Split pre_embedding_lookup
RandyInterfish Mar 15, 2022
84209bd
Update: python a[i
RandyInterfish Mar 16, 2022
78dab1d
Merge branch 'main' into features/gpu_embedding_fusion
RandyInterfish Mar 16, 2022
8c5b929
Merge branch 'features/gpu_embedding_fusion' into features/gpu_embedd…
RandyInterfish Mar 16, 2022
572c7d8
Update: modifying partition_select op
RandyInterfish Mar 17, 2022
c449de3
Update: modify partition_select
RandyInterfish Mar 18, 2022
22506c5
3rd version. Code modifying complete. No compile yet
RandyInterfish Mar 18, 2022
85fc956
Update: compilee pass
RandyInterfish Mar 18, 2022
04d4e60
add more partition strategies
RandyInterfish Mar 19, 2022
241e086
Add one test
RandyInterfish Mar 19, 2022
75a3bd4
Add more unit tests
RandyInterfish Mar 19, 2022
b738015
post op ut passed
RandyInterfish Mar 19, 2022
9ce8da1
ut all passed
RandyInterfish Mar 19, 2022
cf96164
optimize prune and fill moore
RandyInterfish Mar 19, 2022
99cd77e
Minor fixed
RandyInterfish Mar 20, 2022
e9115c3
Update: fix bug and update perf number for modelzoo
RandyInterfish Mar 22, 2022
091748e
Merge branch 'main' of https://github.com/nvzhou/DeepRec-public into …
RandyInterfish Mar 30, 2022
d59c9ce
update api def and golden
RandyInterfish Mar 30, 2022
3ecff7f
Update doc
RandyInterfish Apr 7, 2022
6a13c5b
Ajust interface to V2
RandyInterfish Apr 21, 2022
896d87e
Merge branch 'main' into features/gpu_embedding_fusion
RandyInterfish Apr 26, 2022
901fa8a
Make embedding fusion v1 and v2 compatible
RandyInterfish Apr 26, 2022
132b2ae
Update: delete comments
RandyInterfish Jun 28, 2022
400e0bc
Merge branch 'main' of https://github.com/alibaba/DeepRec into featur…
RandyInterfish Jun 28, 2022
676a324
temp
RandyInterfish Jul 1, 2022
18c62ad
prune and fill with sparse weight seems fine
RandyInterfish Jul 4, 2022
6daa4f4
added sparse_weight to postlookup and grad
RandyInterfish Jul 5, 2022
adf5dae
sparse_weight seems okay
RandyInterfish Jul 13, 2022
d352bfc
Merge branch 'main' of https://github.com/alibaba/DeepRec into featur…
RandyInterfish Jul 13, 2022
14a680b
modify modelzoo
RandyInterfish Jul 15, 2022
aa9f524
Fix unique_with_counts pre-volta hang issue
RandyInterfish Jul 29, 2022
4dfc639
Merge branch 'main' of https://github.com/alibaba/DeepRec into featur…
RandyInterfish Sep 6, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Merge branch 'main' of https://github.com/alibaba/DeepRec into featur… 
…es/gpu_embedding_fusion
  • Loading branch information
@RandyInterfish
RandyInterfish committed Jul 13, 2022
commit d352bfc020e7a1daa10fcb2b2c6350084be53c81
32 changes: 10 additions & 22 deletions tensorflow/tools/api/golden/v1/tensorflow.pbtxt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1457,20 +1457,20 @@ tf_module {
argspec: "args=[\'func\', \'input_signature\', \'autograph\', \'experimental_autograph_options\', \'experimental_relax_shapes\', \'experimental_compile\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'True\', \'None\', \'False\', \'None\'], "
}
member_method {
name: "fused_embedding_sparse_post_look_up"
argspec: "args=[\'emb_shards\', \'partitioned_indices\', \'sp_dense_shape\', \'row_empty_and_invalid_flags\', \'partitioned_values\', \'combiner\', \'default_id\', \'partition_axis\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'-1\', \'0\', \'-1\', \'None\'], "
name: "fused_embedding_local_sparse_look_up"
argspec: "args=[\'sp_values\', \'sp_indices\', \'sp_dense_shape\', \'emb_variable\', \'combiner\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'-1\', \'None\'], "
}
member_method {
name: "fused_embedding_sparse_post_look_up_grad"
argspec: "args=[\'top_grad\', \'emb_shards\', \'partitioned_indices\', \'feature_nums\', \'row_empty_and_invalid_flags\', \'combiner\', \'partition_axis\', \'default_id\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'-1\', \'-1\', \'None\'], "
name: "fused_embedding_local_sparse_look_up_grad"
argspec: "args=[\'top_grad\', \'emb_variable\', \'sp_values\', \'sp_values_offset\', \'combiner\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'-1\', \'None\'], "
}
member_method {
name: "fused_embedding_sparse_post_look_up_v2"
argspec: "args=[\'emb_shards\', \'partition_permutation\', \'sp_dense_shape\', \'indices_before_unique\', \'is_row_empty\', \'unique_idxs\', \'sp_weights_values\', \'combiner\', \'fill_empty_row\', \'default_id\', \'partition_axis\', \'max_norm\', \'use_sparse_weights\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'-1\', \'0\', \'-1\', \'False\', \'None\'], "
name: "fused_embedding_sparse_post_look_up"
argspec: "args=[\'emb_shards\', \'partitioned_indices\', \'sp_dense_shape\', \'row_empty_and_invalid_flags\', \'partitioned_values\', \'combiner\', \'default_id\', \'partition_axis\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'-1\', \'0\', \'-1\', \'None\'], "
}
member_method {
name: "fused_embedding_sparse_post_look_up_v2_grad"
argspec: "args=[\'top_grad\', \'emb_shards\', \'emb_shard_ptrs\', \'partition_permutation\', \'feature_nums\', \'indices_before_unique\', \'unique_idxs\', \'is_row_empty\', \'sp_weights_values\', \'combiner\', \'fill_empty_row\', \'partition_axis\', \'default_id\', \'max_norm\', \'use_sparse_weights\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'0\', \'-1\', \'-1\', \'False\', \'None\'], "
name: "fused_embedding_sparse_post_look_up_grad"
argspec: "args=[\'top_grad\', \'emb_shards\', \'partitioned_indices\', \'feature_nums\', \'row_empty_and_invalid_flags\', \'combiner\', \'partition_axis\', \'default_id\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'-1\', \'-1\', \'None\'], "
}
member_method {
name: "fused_embedding_sparse_pre_look_up"
Expand Down Expand Up @@ -1718,7 +1718,7 @@ tf_module {
}
member_method {
name: "initialize_kv_variable_op"
argspec: "args=[\'resource_self\', \'resource_primary\', \'value\', \'empty_key\', \'shape\', \'counter_type\', \'slot_num\', \'initial_num_buckets\', \'max_load_factor\', \'steps_to_live\', \'ht_type\', \'emb_index\', \'block_num\', \'slot_index\', \'ht_partition_num\', \'filter_freq\', \'max_freq\', \'max_element_size\', \'false_positive_probability\', \'l2_weight_threshold\', \'layout\', \'storage_type\', \'storage_path\', \'storage_size\', \'default_value_dim\', \'record_freq\', \'record_version\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'131072\', \'0.8\', \'0\', \'\', \'0\', \'1\', \'0\', \'1000\', \'0\', \'999999\', \'0\', \'-1\', \'-1\', \'normal\', \'1\', \'.\', \'[]\', \'4096\', \'False\', \'False\', \'None\'], "
argspec: "args=[\'resource_self\', \'resource_primary\', \'value\', \'empty_key\', \'shape\', \'counter_type\', \'slot_num\', \'initial_num_buckets\', \'max_load_factor\', \'steps_to_live\', \'ht_type\', \'emb_index\', \'block_num\', \'slot_index\', \'ht_partition_num\', \'filter_freq\', \'max_freq\', \'max_element_size\', \'false_positive_probability\', \'l2_weight_threshold\', \'layout\', \'storage_type\', \'storage_path\', \'storage_size\', \'default_value_dim\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'131072\', \'0.8\', \'0\', \'\', \'0\', \'1\', \'0\', \'1000\', \'0\', \'999999\', \'0\', \'-1\', \'-1\', \'normal\', \'1\', \'.\', \'[]\', \'4096\', \'None\'], "
}
member_method {
name: "initialize_local_variables"
Expand Down Expand Up @@ -1830,7 +1830,7 @@ tf_module {
}
member_method {
name: "kv_resource_import_v2"
argspec: "args=[\'prefix\', \'resource_self\', \'resource_primary\', \'value\', \'tensor_names\', \'empty_key\', \'shape\', \'counter_type\', \'slot_num\', \'emb_index\', \'slot_index\', \'block_num\', \'steps_to_live\', \'partition_id\', \'partition_num\', \'ht_type\', \'filter_freq\', \'ht_partition_num\', \'max_element_size\', \'false_positive_probability\', \'l2_weight_threshold\', \'layout\', \'max_freq\', \'storage_type\', \'storage_path\', \'storage_size\', \'default_value_dim\', \'record_freq\', \'record_version\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'0\', \'1\', \'0\', \'0\', \'1\', \'\', \'0\', \'1000\', \'0\', \'-1\', \'-1\', \'normal\', \'999999\', \'1\', \'.\', \'[]\', \'4096\', \'False\', \'False\', \'None\'], "
argspec: "args=[\'prefix\', \'resource_self\', \'resource_primary\', \'value\', \'tensor_names\', \'empty_key\', \'shape\', \'counter_type\', \'slot_num\', \'emb_index\', \'slot_index\', \'block_num\', \'steps_to_live\', \'partition_id\', \'partition_num\', \'ht_type\', \'filter_freq\', \'ht_partition_num\', \'max_element_size\', \'false_positive_probability\', \'l2_weight_threshold\', \'layout\', \'max_freq\', \'storage_type\', \'storage_path\', \'storage_size\', \'default_value_dim\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'0\', \'1\', \'0\', \'0\', \'1\', \'\', \'0\', \'1000\', \'0\', \'-1\', \'-1\', \'normal\', \'999999\', \'1\', \'.\', \'[]\', \'4096\', \'None\'], "
}
member_method {
name: "kv_resource_incr_import"
Expand Down Expand Up @@ -2140,10 +2140,6 @@ tf_module {
name: "parse_tensor"
argspec: "args=[\'serialized\', \'out_type\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "partition_with_permutation"
argspec: "args=[\'input\', \'partition_shapes\', \'partition_strategy\', \'partition_axis\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'None\'], "
}
member_method {
name: "placeholder"
argspec: "args=[\'dtype\', \'shape\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
Expand All @@ -2168,10 +2164,6 @@ tf_module {
name: "print"
argspec: "args=[], varargs=inputs, keywords=kwargs, defaults=None"
}
member_method {
name: "prune_invalid_and_fill_empty_rows"
argspec: "args=[\'sp_values\', \'sp_indices\', \'sp_dense_shape\', \'sp_weights_values\', \'fill_empty_row\', \'prune\', \'default_id\', \'use_sparse_weights\', \'prune_sparse_weights\', \'default_weight\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'False\', \'-1\', \'False\', \'False\', \'1\', \'None\'], "
}
member_method {
name: "py_func"
argspec: "args=[\'func\', \'inp\', \'Tout\', \'stateful\', \'name\'], varargs=None, keywords=None, defaults=[\'True\', \'None\'], "
Expand Down Expand Up @@ -3020,10 +3012,6 @@ tf_module {
name: "unique_with_counts"
argspec: "args=[\'x\', \'out_idx\', \'name\'], varargs=None, keywords=None, defaults=[\"<dtype: \'int32\'>\", \'None\'], "
}
member_method {
name: "unique_with_counts_v3"
argspec: "args=[\'input\', \'CounterType\', \'name\'], varargs=None, keywords=None, defaults=[\"<dtype: \'int32\'>\", \'None\'], "
}
member_method {
name: "unravel_index"
argspec: "args=[\'indices\', \'dims\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
Expand Down
32 changes: 10 additions & 22 deletions tensorflow/tools/api/golden/v1/tensorflow.raw_ops.pbtxt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1593,20 +1593,20 @@ tf_module {
argspec: "args=[\'x\', \'scale\', \'offset\', \'mean\', \'variance\', \'epsilon\', \'exponential_avg_factor\', \'data_format\', \'is_training\', \'name\'], varargs=None, keywords=None, defaults=[\'0.0001\', \'1\', \'NHWC\', \'True\', \'None\'], "
}
member_method {
name: "FusedEmbeddingSparsePostLookUp"
argspec: "args=[\'emb_shards\', \'partitioned_indices\', \'sp_dense_shape\', \'row_empty_and_invalid_flags\', \'partitioned_values\', \'combiner\', \'default_id\', \'partition_axis\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'-1\', \'0\', \'-1\', \'None\'], "
name: "FusedEmbeddingLocalSparseLookUp"
argspec: "args=[\'sp_values\', \'sp_indices\', \'sp_dense_shape\', \'emb_variable\', \'combiner\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'-1\', \'None\'], "
}
member_method {
name: "FusedEmbeddingSparsePostLookUpGrad"
argspec: "args=[\'top_grad\', \'emb_shards\', \'partitioned_indices\', \'feature_nums\', \'row_empty_and_invalid_flags\', \'combiner\', \'partition_axis\', \'default_id\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'-1\', \'-1\', \'None\'], "
name: "FusedEmbeddingLocalSparseLookUpGrad"
argspec: "args=[\'top_grad\', \'emb_variable\', \'sp_values\', \'sp_values_offset\', \'combiner\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'-1\', \'None\'], "
}
member_method {
name: "FusedEmbeddingSparsePostLookUpV2"
argspec: "args=[\'emb_shards\', \'partition_permutation\', \'sp_dense_shape\', \'indices_before_unique\', \'is_row_empty\', \'unique_idxs\', \'sp_weights_values\', \'combiner\', \'fill_empty_row\', \'default_id\', \'partition_axis\', \'max_norm\', \'use_sparse_weights\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'-1\', \'0\', \'-1\', \'False\', \'None\'], "
name: "FusedEmbeddingSparsePostLookUp"
argspec: "args=[\'emb_shards\', \'partitioned_indices\', \'sp_dense_shape\', \'row_empty_and_invalid_flags\', \'partitioned_values\', \'combiner\', \'default_id\', \'partition_axis\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'-1\', \'0\', \'-1\', \'None\'], "
}
member_method {
name: "FusedEmbeddingSparsePostLookUpV2Grad"
argspec: "args=[\'top_grad\', \'emb_shards\', \'emb_shard_ptrs\', \'partition_permutation\', \'feature_nums\', \'indices_before_unique\', \'unique_idxs\', \'is_row_empty\', \'sp_weights_values\', \'combiner\', \'fill_empty_row\', \'partition_axis\', \'default_id\', \'max_norm\', \'use_sparse_weights\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'0\', \'-1\', \'-1\', \'False\', \'None\'], "
name: "FusedEmbeddingSparsePostLookUpGrad"
argspec: "args=[\'top_grad\', \'emb_shards\', \'partitioned_indices\', \'feature_nums\', \'row_empty_and_invalid_flags\', \'combiner\', \'partition_axis\', \'default_id\', \'max_norm\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'-1\', \'-1\', \'None\'], "
}
member_method {
name: "FusedEmbeddingSparsePreLookUp"
Expand Down Expand Up @@ -1930,7 +1930,7 @@ tf_module {
}
member_method {
name: "InitializeKvVariableOp"
argspec: "args=[\'resource_self\', \'resource_primary\', \'value\', \'empty_key\', \'shape\', \'counter_type\', \'slot_num\', \'initial_num_buckets\', \'max_load_factor\', \'steps_to_live\', \'ht_type\', \'emb_index\', \'block_num\', \'slot_index\', \'ht_partition_num\', \'filter_freq\', \'max_freq\', \'max_element_size\', \'false_positive_probability\', \'l2_weight_threshold\', \'layout\', \'storage_type\', \'storage_path\', \'storage_size\', \'default_value_dim\', \'record_freq\', \'record_version\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'131072\', \'0.8\', \'0\', \'\', \'0\', \'1\', \'0\', \'1000\', \'0\', \'999999\', \'0\', \'-1\', \'-1\', \'normal\', \'1\', \'.\', \'[]\', \'4096\', \'False\', \'False\', \'None\'], "
argspec: "args=[\'resource_self\', \'resource_primary\', \'value\', \'empty_key\', \'shape\', \'counter_type\', \'slot_num\', \'initial_num_buckets\', \'max_load_factor\', \'steps_to_live\', \'ht_type\', \'emb_index\', \'block_num\', \'slot_index\', \'ht_partition_num\', \'filter_freq\', \'max_freq\', \'max_element_size\', \'false_positive_probability\', \'l2_weight_threshold\', \'layout\', \'storage_type\', \'storage_path\', \'storage_size\', \'default_value_dim\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'131072\', \'0.8\', \'0\', \'\', \'0\', \'1\', \'0\', \'1000\', \'0\', \'999999\', \'0\', \'-1\', \'-1\', \'normal\', \'1\', \'.\', \'[]\', \'4096\', \'None\'], "
}
member_method {
name: "InitializeTable"
Expand Down Expand Up @@ -2066,7 +2066,7 @@ tf_module {
}
member_method {
name: "KvResourceImportV2"
argspec: "args=[\'prefix\', \'resource_self\', \'resource_primary\', \'value\', \'tensor_names\', \'empty_key\', \'shape\', \'counter_type\', \'slot_num\', \'emb_index\', \'slot_index\', \'block_num\', \'steps_to_live\', \'partition_id\', \'partition_num\', \'ht_type\', \'filter_freq\', \'ht_partition_num\', \'max_element_size\', \'false_positive_probability\', \'l2_weight_threshold\', \'layout\', \'max_freq\', \'storage_type\', \'storage_path\', \'storage_size\', \'default_value_dim\', \'record_freq\', \'record_version\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'0\', \'1\', \'0\', \'0\', \'1\', \'\', \'0\', \'1000\', \'0\', \'-1\', \'-1\', \'normal\', \'999999\', \'1\', \'.\', \'[]\', \'4096\', \'False\', \'False\', \'None\'], "
argspec: "args=[\'prefix\', \'resource_self\', \'resource_primary\', \'value\', \'tensor_names\', \'empty_key\', \'shape\', \'counter_type\', \'slot_num\', \'emb_index\', \'slot_index\', \'block_num\', \'steps_to_live\', \'partition_id\', \'partition_num\', \'ht_type\', \'filter_freq\', \'ht_partition_num\', \'max_element_size\', \'false_positive_probability\', \'l2_weight_threshold\', \'layout\', \'max_freq\', \'storage_type\', \'storage_path\', \'storage_size\', \'default_value_dim\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'0\', \'1\', \'0\', \'0\', \'1\', \'\', \'0\', \'1000\', \'0\', \'-1\', \'-1\', \'normal\', \'999999\', \'1\', \'.\', \'[]\', \'4096\', \'None\'], "
}
member_method {
name: "KvResourceIncrImport"
Expand Down Expand Up @@ -2856,10 +2856,6 @@ tf_module {
name: "ParseTensor"
argspec: "args=[\'serialized\', \'out_type\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "PartitionWithPermutation"
argspec: "args=[\'input\', \'partition_shapes\', \'partition_strategy\', \'partition_axis\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'None\'], "
}
member_method {
name: "PartitionedCall"
argspec: "args=[\'args\', \'Tout\', \'f\', \'config\', \'config_proto\', \'executor_type\', \'name\'], varargs=None, keywords=None, defaults=[\'\', \'\', \'\', \'None\'], "
Expand Down Expand Up @@ -2928,10 +2924,6 @@ tf_module {
name: "Prod"
argspec: "args=[\'input\', \'axis\', \'keep_dims\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'None\'], "
}
member_method {
name: "PruneInvalidAndFillEmptyRows"
argspec: "args=[\'sp_values\', \'sp_indices\', \'sp_dense_shape\', \'sp_weights_values\', \'fill_empty_row\', \'prune\', \'default_id\', \'use_sparse_weights\', \'prune_sparse_weights\', \'default_weight\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'False\', \'-1\', \'False\', \'False\', \'1\', \'None\'], "
}
member_method {
name: "PyFunc"
argspec: "args=[\'input\', \'token\', \'Tout\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
Expand Down Expand Up @@ -5192,10 +5184,6 @@ tf_module {
name: "UniqueWithCountsV2"
argspec: "args=[\'x\', \'axis\', \'out_idx\', \'name\'], varargs=None, keywords=None, defaults=[\"<dtype: \'int32\'>\", \'None\'], "
}
member_method {
name: "UniqueWithCountsV3"
argspec: "args=[\'input\', \'CounterType\', \'name\'], varargs=None, keywords=None, defaults=[\"<dtype: \'int32\'>\", \'None\'], "
}
member_method {
name: "Unpack"
argspec: "args=[\'value\', \'num\', \'axis\', \'name\'], varargs=None, keywords=None, defaults=[\'0\', \'None\'], "
Expand Down
Loading
You are viewing a condensed version of this merge commit. You can view the full changes here.