Home
last modified time | relevance | path

Searched refs:to_retain (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py1662 def sparse_retain(sp_input, to_retain): argument
1691 to_retain = ops.convert_to_tensor(to_retain)
1694 retain_shape = to_retain.get_shape()
1700 where_true = array_ops.reshape(array_ops.where(to_retain), [-1])
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_ops_test.py303 to_retain = np.array([1, 0, 0, 1, 1, 0], dtype=np.bool)
304 sp_output = sparse_ops.sparse_retain(sp_input, to_retain)
315 to_retain = np.zeros((6,), dtype=np.bool)
316 sp_output = sparse_ops.sparse_retain(sp_input, to_retain)
327 to_retain = np.array([1, 0, 0, 1, 0], dtype=np.bool)
329 sparse_ops.sparse_retain(sp_input, to_retain)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.sparse.pbtxt73 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.sparse.pbtxt93 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.pbtxt2129 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"