Searched refs:id_weight_pair (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/python/feature_column/ |
D | sequence_feature_column_test.py | 177 id_weight_pair = _get_sparse_tensors(column, {'aaa': inputs}) 179 self.assertIsNone(id_weight_pair.weight_tensor) 181 self, expected, self.evaluate(id_weight_pair.id_tensor)) 252 id_weight_pair = _get_sparse_tensors(column, {'aaa': inputs}) 254 self.assertIsNone(id_weight_pair.weight_tensor) 256 self, expected, self.evaluate(id_weight_pair.id_tensor)) 305 id_weight_pair = _get_sparse_tensors(column, {'aaa': inputs}) 307 self.assertIsNone(id_weight_pair.weight_tensor) 311 self, expected, self.evaluate(id_weight_pair.id_tensor)) 331 id_weight_pair = _get_sparse_tensors(column, {'aaa': input_placeholder}) [all …]
|
D | feature_column_v2_test.py | 614 id_weight_pair = bucketized_price.get_sparse_tensors( 616 self.assertIsNone(id_weight_pair.weight_tensor) 617 id_tensor_value = sess.run(id_weight_pair.id_tensor) 632 id_weight_pair = bucketized_price.get_sparse_tensors( 634 self.assertIsNone(id_weight_pair.weight_tensor) 635 id_tensor_value = sess.run(id_weight_pair.id_tensor) 944 id_weight_pair = hashed_sparse.get_sparse_tensors(transformation_cache, 946 self.assertIsNone(id_weight_pair.weight_tensor) 948 transformation_cache.get(hashed_sparse, None), id_weight_pair.id_tensor) 955 id_weight_pair = hashed_sparse.get_sparse_tensors(transformation_cache, [all …]
|
D | feature_column_test.py | 499 id_weight_pair = bucketized_price._get_sparse_tensors(builder) 500 self.assertIsNone(id_weight_pair.weight_tensor) 501 id_tensor_value = sess.run(id_weight_pair.id_tensor) 514 id_weight_pair = bucketized_price._get_sparse_tensors(builder) 515 self.assertIsNone(id_weight_pair.weight_tensor) 516 id_tensor_value = sess.run(id_weight_pair.id_tensor) 823 id_weight_pair = hashed_sparse._get_sparse_tensors(builder) 824 self.assertIsNone(id_weight_pair.weight_tensor) 825 self.assertEqual(builder.get(hashed_sparse), id_weight_pair.id_tensor) 845 id_weight_pair = hashed_sparse._get_sparse_tensors(builder) [all …]
|
D | feature_column_v2.py | 4085 def _transform_id_weight_pair(self, id_weight_pair, size): argument 4086 id_tensor = id_weight_pair.id_tensor 4087 weight_tensor = id_weight_pair.weight_tensor 4128 id_weight_pair = self.categorical_column.get_sparse_tensors( 4130 return self._transform_id_weight_pair(id_weight_pair, 4136 …id_weight_pair = self.categorical_column._get_sparse_tensors(inputs) # pylint: disable=protected-… 4137 return self._transform_id_weight_pair(id_weight_pair,
|
D | feature_column.py | 3125 …id_weight_pair = self.categorical_column._get_sparse_tensors(inputs) # pylint: disable=protected-… 3126 id_tensor = id_weight_pair.id_tensor 3127 weight_tensor = id_weight_pair.weight_tensor
|