Home
last modified time | relevance | path

Searched refs:sparse_retain (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py996 sparse_ids = sparse_ops.sparse_retain(sparse_ids, is_id_valid)
998 sparse_weights = sparse_ops.sparse_retain(sparse_weights, is_id_valid)
1006 sparse_ids = sparse_ops.sparse_retain(sparse_ids, is_weights_valid)
1007 sparse_weights = sparse_ops.sparse_retain(sparse_weights, is_weights_valid)
Dsparse_ops.py1948 def sparse_retain(sp_input, to_retain): function
Dmetrics_impl.py2248 return sparse_ops.sparse_retain(ids, math_ops.equal(ids.values,
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_ops_test.py304 sp_output = sparse_ops.sparse_retain(sp_input, to_retain)
316 sp_output = sparse_ops.sparse_retain(sp_input, to_retain)
329 sparse_ops.sparse_retain(sp_input, to_retain)
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py4047 sparse_ids = sparse_ops.sparse_retain(sparse_ids, is_id_valid)
4049 sparse_weights = sparse_ops.sparse_retain(sparse_weights, is_id_valid)
4057 sparse_ids = sparse_ops.sparse_retain(sparse_ids, is_weights_valid)
4058 sparse_weights = sparse_ops.sparse_retain(sparse_weights, is_weights_valid)
/external/tensorflow/tensorflow/python/training/
Dinput.py548 t = sparse_ops.sparse_retain(t, _sparse_values_to_keep(t, keep_input))
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2212 name: "sparse_retain"