Lines Matching refs:features
31 def model_builder(features, labels, mode, params, config):
35 features: `Tensor` or `dict` of `Tensor` objects.
64 if features is None:
72 if isinstance(features, ops.Tensor):
73 features = {features.name: features}
75 # Make a shallow copy of features to ensure downstream usage
77 training_features = copy.copy(features)
96 features=training_features)
101 logits = logits_modifier_function(logits, features, mode)
112 features=features,
