Lines Matching refs:features
31 def dummy_logit_fn(features, mode):
33 return features['f1']
35 return features['f2']
36 features = {
41 dummy_logit_fn, features, model_fn.ModeKeys.EVAL, 'fake_params',
48 def dummy_logit_fn(features):
49 return {u'head1': features['f1'], 'head2': features['f2']}
51 features = {
55 logit_fn_result = logit_fns.call_logit_fn(dummy_logit_fn, features,
64 def invalid_logit_fn(features, params):
66 features['f1'] * params['input_multiplier'],
67 features['f2'] * params['input_multiplier']
70 features = {
78 logit_fns.call_logit_fn(invalid_logit_fn, features, 'fake_mode', params,
83 def invalid_logit_fn(features):
84 return {'head1': features['f1'], 'head2': features['f2']}
86 features = {'f1': constant_op.constant([[2., 3.]]), 'f2': 'some string'}
90 logit_fns.call_logit_fn(invalid_logit_fn, features, 'fake_mode',
