Searched refs:feature_key (Results 1 – 2 of 2) sorted by relevance
798 def _add_batched_ragged_partition(rt, partition, tensor_dict, feature_key, argument843 % feature_key))
866 feature_key = (feature_tag, lookup_indices)867 feature_index = feature_indices.get(feature_key)877 feature_indices[feature_key] = feature_index