Searched refs:cols_to_vars (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util_test.py | 101 cols_to_vars = {}
108 cols_to_vars=cols_to_vars)
110 return cols_to_vars
112 def _assert_cols_to_vars(self, cols_to_vars, cols_to_expected_values, sess): argument
114 for i, var in enumerate(cols_to_vars[col]):
544 cols_to_vars = self._create_linear_model([sc_int], partitioner)
548 self._assert_cols_to_vars(cols_to_vars, {sc_int: [np.zeros([10, 1])]},
554 cols_to_vars = self._create_linear_model([sc_int], partitioner)
558 self._assert_cols_to_vars(cols_to_vars, {sc_int: [prev_int_val]}, sess)
573 cols_to_vars = self._create_linear_model([sc_hash], partitioner)
[all …]
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_test.py | 1320 cols_to_vars=None): argument
1329 if cols_to_vars is not None:
1330 cols_to_vars.update(keras_linear_model.cols_to_vars())
1656 cols_to_vars = {}
1657 fc.linear_model(features, [price1, price2], cols_to_vars=cols_to_vars)
1661 self.assertAllEqual(cols_to_vars['bias'], [bias])
1662 self.assertAllEqual(cols_to_vars[price1], [price1_var])
1663 self.assertAllEqual(cols_to_vars[price2], [price2_var])
1673 cols_to_vars = {}
1677 fc.linear_model(features, [price1, price2], cols_to_vars=cols_to_vars)
[all …]
|
| D | feature_column.py | 175 cols_to_vars=None, argument
211 if cols_to_vars is not None:
214 cols_to_vars[column] = ops.get_collection(
238 cols_to_vars=None, argument
301 cols_to_vars=cols_to_vars,
316 cols_to_vars=None, argument
324 self._cols_to_vars = cols_to_vars
336 cols_to_vars=None,
375 cols_to_vars=None): argument
496 if cols_to_vars is not None:
[all …]
|
| D | feature_column_v2_test.py | 2656 cols_to_vars = {}
2657 fc_old.linear_model(features, [price1, price2], cols_to_vars=cols_to_vars)
2661 self.assertAllEqual(cols_to_vars['bias'], [bias])
2662 self.assertAllEqual(cols_to_vars[price1], [price1_var])
2663 self.assertAllEqual(cols_to_vars[price2], [price2_var])
2673 cols_to_vars = {}
2678 features, [price1, price2], cols_to_vars=cols_to_vars)
2683 self.assertEqual([0.], self.evaluate(cols_to_vars['bias'][0]))
2685 self.assertAllEqual([[0.]], self.evaluate(cols_to_vars[price1][0]))
2686 self.assertAllEqual([[0.]], self.evaluate(cols_to_vars[price1][1]))
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.feature_column.pbtxt | 37 …ures\', \'feature_columns\', \'weight_collections\', \'trainable\', \'cols_to_vars\', \'cols_to_ou…
41 …nits\', \'sparse_combiner\', \'weight_collections\', \'trainable\', \'cols_to_vars\'], varargs=Non…
|