Home | History | Annotate | Download | only in estimators

Lines Matching refs:features

180     features: `Tensor` or `dict` of `Tensor` objects.
188 features = parsing_ops.parse_example(examples_batch, feature_spec)
191 features = features[single_feature_name]
192 if not signatures.dtype.is_compatible_with(features.dtype):
193 features = math_ops.cast(features, signatures.dtype)
194 return features
195 # Returns dict of features, casts if needed.
196 for name in features:
197 if not signatures[name].dtype.is_compatible_with(features[name].dtype):
198 features[name] = math_ops.cast(features[name], signatures[name].dtype)
199 return features