Searched refs:feature_key (Results 1 – 2 of 2) sorted by relevance
804 def _add_batched_ragged_partition(rt, partition, tensor_dict, feature_key, argument849 % feature_key))
892 feature_key = (feature_tag, lookup_indices)893 feature_index = feature_indices.get(feature_key)903 feature_indices[feature_key] = feature_index