Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
Dsparse_ops_test.py299 to_retain = np.array([1, 0, 0, 1, 1, 0], dtype=np.bool_)
300 sp_output = sparse_ops.sparse_retain(sp_input, to_retain)
311 to_retain = np.zeros((6,), dtype=np.bool_)
312 sp_output = sparse_ops.sparse_retain(sp_input, to_retain)
323 to_retain = np.array([1, 0, 0, 1, 0], dtype=np.bool_)
325 sparse_ops.sparse_retain(sp_input, to_retain)
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py1941 def sparse_retain(sp_input, to_retain): argument
1970 to_retain = ops.convert_to_tensor(to_retain)
1973 retain_shape = to_retain.get_shape()
1979 where_true = array_ops.reshape(array_ops.where_v2(to_retain), [-1])
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.sparse.pbtxt81 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.sparse.pbtxt101 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"
Dtensorflow.pbtxt2221 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"