Home
last modified time | relevance | path

Searched refs:weight_feature_key (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py1660 weight_feature_key, argument
1730 weight_feature_key=weight_feature_key,
3772 self.categorical_column.name, self.weight_feature_key)
3778 if self.weight_feature_key in config:
3780 config[self.weight_feature_key], self.weight_feature_key))
3781 config[self.weight_feature_key] = parsing_ops.VarLenFeature(self.dtype)
3789 if self.weight_feature_key in config:
3791 config[self.weight_feature_key], self.weight_feature_key))
3792 config[self.weight_feature_key] = parsing_ops.VarLenFeature(self.dtype)
3808 raise ValueError('Missing weights {}.'.format(self.weight_feature_key))
[all …]
Dfeature_column.py1497 weight_feature_key, argument
1567 weight_feature_key=weight_feature_key,
2985 self.categorical_column.name, self.weight_feature_key)
2990 if self.weight_feature_key in config:
2992 config[self.weight_feature_key], self.weight_feature_key))
2993 config[self.weight_feature_key] = parsing_ops.VarLenFeature(self.dtype)
3001 weight_tensor = inputs.get(self.weight_feature_key)
3003 raise ValueError('Missing weights {}.'.format(self.weight_feature_key))
Dfeature_column_v2_test.py5600 categorical_column_a, weight_feature_key='aaa_weights')
5604 categorical_column_b, weight_feature_key='bbb_weights')
6007 weight_feature_key='values')
6020 weight_feature_key='values')
6028 weight_feature_key='values')
6042 weight_feature_key='values',
6050 weight_feature_key='values',
6057 weight_feature_key='values')
6073 weight_feature_key='aaa').parse_example_spec()
6079 weight_feature_key='values')
[all …]
Dfeature_column_test.py5679 categorical_column_a, weight_feature_key='aaa_weights')
5683 categorical_column_b, weight_feature_key='bbb_weights')
6273 weight_feature_key='values')
6287 weight_feature_key='values')
6301 weight_feature_key='values',
6309 weight_feature_key='values',
6316 weight_feature_key='values')
6329 weight_feature_key='aaa')._parse_example_spec()
6335 weight_feature_key='values')
6349 a, weight_feature_key='weights')
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.feature_column.pbtxt69 …argspec: "args=[\'categorical_column\', \'weight_feature_key\', \'dtype\'], varargs=None, keywords…
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column.py433 return self.categorical_column.weight_feature_key
572 return self.categorical_column.weight_feature_key
Dfeature_column_v2.py489 return self.categorical_column.weight_feature_key
683 return self.categorical_column.weight_feature_key
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.feature_column.pbtxt77 …argspec: "args=[\'categorical_column\', \'weight_feature_key\', \'dtype\'], varargs=None, keywords…