Searched refs:sp_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | utils.py | 30 def sequence_length_from_sparse_tensor(sp_tensor, num_elements=1): argument
33 row_ids = sp_tensor.indices[:, 0]
34 column_ids = sp_tensor.indices[:, 1]
50 n_pad = array_ops.shape(sp_tensor)[:1] - array_ops.shape(seq_length)[:1]
|
| D | sequence_feature_column.py | 441 sp_tensor = transformation_cache.get(self, state_manager)
443 sp_tensor, default_value=self.default_value)
454 if sp_tensor.shape.ndims == 2:
459 sp_tensor, num_elements=num_elements)
|
| D | feature_column_test.py | 1432 sp_tensor = sparse_tensor.SparseTensor(
1436 return _CategoricalColumn.IdWeightPair(sp_tensor, None)
1440 sp_tensor = sparse_tensor.SparseTensor(
1444 features = {dense_and_sparse_column.name: sp_tensor}
2117 sp_tensor = sparse_tensor.SparseTensor(
2121 return _CategoricalColumn.IdWeightPair(sp_tensor, None)
2125 sp_tensor = sparse_tensor.SparseTensor(
2129 features = {dense_and_sparse_column.name: sp_tensor}
|
| D | feature_column_v2_test.py | 1573 sp_tensor = sparse_tensor.SparseTensor(
1577 return fc.CategoricalColumn.IdWeightPair(sp_tensor, None)
1581 sp_tensor = sparse_tensor.SparseTensor(
1585 features = {dense_and_sparse_column.name: sp_tensor}
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 217 def from_sparse_tensor(sp_tensor, weights=None): argument
219 sp_tensor.values,
220 sp_tensor.indices,
|