Searched refs:ragged_ids (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 944 ragged_ids, argument
972 if ragged_ids.dtype != dtypes.int32 and ragged_ids.dtype != dtypes.int64:
974 str(ragged_ids.dtype) +
982 ids=ragged_ids,
|
| D | nn_test.py | 1752 ragged_ids = ragged_factory_ops.constant([[1, 2, 3], [0], [1, 2]],
1755 embedded_ragged = nn.embedding_lookup_ragged(weights, ragged_ids)
1766 ragged_ids = ragged_factory_ops.constant(
1771 embedded_ragged = nn.embedding_lookup_ragged(weights, ragged_ids)
1781 ragged_ids = ragged_factory_ops.constant([[1, 2, 3], [0], [1, 2]])
1785 nn.embedding_lookup_ragged(None, ragged_ids)
1788 ragged_ids = ragged_factory_ops.constant([[1, 2, 3], [0], [1, 2]])
1792 nn.embedding_lookup_ragged([], ragged_ids)
1796 ragged_ids = ragged_factory_ops.constant([[1., 2., 3.], [1., 2.]])
1800 nn.embedding_lookup_ragged(weights, ragged_ids)
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | lookup_ops_test.py | 1166 ragged_ids = table.lookup(ragged_features)
1168 self.assertAllEqual([5], ragged_ids.values._shape_as_list())
1171 [ragged_ids.values, ragged_ids.row_splits])
1215 ragged_ids = table.lookup(ragged_features)
1217 self.assertAllEqual([5], ragged_ids.values._shape_as_list())
1220 [ragged_ids.values, ragged_ids.row_splits])
1258 ragged_ids = table.lookup(ragged_features)
1260 self.assertAllEqual([5], ragged_ids.values._shape_as_list())
1263 [ragged_ids.values, ragged_ids.row_splits])
2861 ragged_ids = table.lookup(ragged_features)
[all …]
|