Searched refs:to_retain (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/python/ops/ |
D | sparse_ops.py | 1662 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/ |
D | sparse_ops_test.py | 303 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/ |
D | tensorflow.sparse.pbtxt | 73 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.sparse.pbtxt | 93 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"
|
D | tensorflow.pbtxt | 2129 argspec: "args=[\'sp_input\', \'to_retain\'], varargs=None, keywords=None, defaults=None"
|