Searched refs:sparse_embedding_aggregate_slice (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | feature_column_v2.py | 791 def sparse_embedding_aggregate_slice(params, function
975 return sparse_embedding_aggregate_slice(embedding_weights, (values, mask),
1020 return sparse_embedding_aggregate_slice(embedding_weights, (values, mask),
1091 return sparse_embedding_aggregate_slice(embedding_weights, (values, mask),
|