Searched refs:cols_to_vars (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util_test.py | 103 cols_to_vars = {}
110 cols_to_vars=cols_to_vars)
112 return cols_to_vars
114 def _assert_cols_to_vars(self, cols_to_vars, cols_to_expected_values, sess): argument
116 for i, var in enumerate(cols_to_vars[col]):
546 cols_to_vars = self._create_linear_model([sc_int], partitioner)
550 self._assert_cols_to_vars(cols_to_vars, {sc_int: [np.zeros([10, 1])]},
556 cols_to_vars = self._create_linear_model([sc_int], partitioner)
560 self._assert_cols_to_vars(cols_to_vars, {sc_int: [prev_int_val]}, sess)
575 cols_to_vars = self._create_linear_model([sc_hash], partitioner)
[all …]
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_test.py | 1292 cols_to_vars=None): argument
1301 if cols_to_vars is not None:
1302 cols_to_vars.update(keras_linear_model.cols_to_vars())
1628 cols_to_vars = {}
1629 fc.linear_model(features, [price1, price2], cols_to_vars=cols_to_vars)
1633 self.assertAllEqual(cols_to_vars['bias'], [bias])
1634 self.assertAllEqual(cols_to_vars[price1], [price1_var])
1635 self.assertAllEqual(cols_to_vars[price2], [price2_var])
1645 cols_to_vars = {}
1649 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
501 if cols_to_vars is not None:
[all …]
|
| D | feature_column_v2_test.py | 1767 cols_to_vars = {}
1768 fc_old.linear_model(features, [price1, price2], cols_to_vars=cols_to_vars)
1772 self.assertAllEqual(cols_to_vars['bias'], [bias])
1773 self.assertAllEqual(cols_to_vars[price1], [price1_var])
1774 self.assertAllEqual(cols_to_vars[price2], [price2_var])
1784 cols_to_vars = {}
1789 features, [price1, price2], cols_to_vars=cols_to_vars)
1794 self.assertEqual([0.], self.evaluate(cols_to_vars['bias'][0]))
1796 self.assertAllEqual([[0.]], self.evaluate(cols_to_vars[price1][0]))
1797 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…
|