Searched refs:sp_tensor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | sequence_queueing_state_saver.py | 1662 def _copy_sparse_tensor(sp_tensor): argument
1675 values = array_ops.tile(sp_tensor.values, array_ops.expand_dims(n, 0))
1677 [array_ops.expand_dims(value_length, 0), sp_tensor.dense_shape], 0)
1682 array_ops.expand_dims(sp_tensor.indices, 0),
1692 dim0 = array_ops.shape(sp_tensor.indices)[0]
1693 dim1 = array_ops.shape(sp_tensor.indices)[1]
1724 sp_tensor = sequences[key]
1726 sp_tensor = sparse_tensor.SparseTensor(
1727 sp_tensor.indices, # with only 0s at column 1 representing time.
1728 sp_tensor.values,
[all …]
|
| /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 | 551 sp_tensor = transformation_cache.get(self, state_manager)
553 sp_tensor, default_value=self.default_value)
564 if sp_tensor.shape.ndims == 2:
569 sp_tensor, num_elements=num_elements)
|
| D | feature_column_test.py | 1460 sp_tensor = sparse_tensor.SparseTensor(
1464 return _CategoricalColumn.IdWeightPair(sp_tensor, None)
1468 sp_tensor = sparse_tensor.SparseTensor(
1472 features = {dense_and_sparse_column.name: sp_tensor}
2145 sp_tensor = sparse_tensor.SparseTensor(
2149 return _CategoricalColumn.IdWeightPair(sp_tensor, None)
2153 sp_tensor = sparse_tensor.SparseTensor(
2157 features = {dense_and_sparse_column.name: sp_tensor}
|
| D | feature_column_v2_test.py | 1731 sp_tensor = sparse_tensor.SparseTensor(
1735 return fc.CategoricalColumn.IdWeightPair(sp_tensor, None)
1739 sp_tensor = sparse_tensor.SparseTensor(
1743 features = {dense_and_sparse_column.name: sp_tensor}
2462 sp_tensor = sparse_tensor.SparseTensor(
2466 return fc.CategoricalColumn.IdWeightPair(sp_tensor, None)
2470 sp_tensor = sparse_tensor.SparseTensor(
2474 features = {dense_and_sparse_column.name: sp_tensor}
|
| /external/tensorflow/tensorflow/contrib/feature_column/python/feature_column/ |
| D | sequence_feature_column.py | 495 sp_tensor = inputs.get(self)
497 sp_tensor, default_value=self.default_value)
509 if sp_tensor.shape.ndims == 2 else 1)
511 sp_tensor, num_elements=num_elements)
|