diff options
author | 2017-07-11 10:06:30 -0700 | |
---|---|---|
committer | 2017-07-11 10:10:59 -0700 | |
commit | dbe029ded02962adf0227ae79a1b0f59e9e8d3dd (patch) | |
tree | 8f171e2e89811754d4ff1cacfc6456d1baed6488 | |
parent | 0478774fcc918cb7a81bfa8832d115b376bfb97b (diff) |
Update ops-related pbtxt files.
PiperOrigin-RevId: 161543495
-rw-r--r-- | tensorflow/core/ops/compat/ops_history.v1.pbtxt | 43 | ||||
-rw-r--r-- | tensorflow/core/ops/ops.pbtxt | 49 |
2 files changed, 92 insertions, 0 deletions
diff --git a/tensorflow/core/ops/compat/ops_history.v1.pbtxt b/tensorflow/core/ops/compat/ops_history.v1.pbtxt index 0198d4ff22..4fcf7aaa41 100644 --- a/tensorflow/core/ops/compat/ops_history.v1.pbtxt +++ b/tensorflow/core/ops/compat/ops_history.v1.pbtxt @@ -10074,6 +10074,49 @@ op { } } op { + name: "GatherV2" + input_arg { + name: "params" + type_attr: "Tparams" + } + input_arg { + name: "indices" + type_attr: "Tindices" + } + input_arg { + name: "axis" + type_attr: "Taxis" + } + output_arg { + name: "output" + type_attr: "Tparams" + } + attr { + name: "Tparams" + type: "type" + } + attr { + name: "Tindices" + type: "type" + allowed_values { + list { + type: DT_INT32 + type: DT_INT64 + } + } + } + attr { + name: "Taxis" + type: "type" + allowed_values { + list { + type: DT_INT32 + type: DT_INT64 + } + } + } +} +op { name: "GetSessionHandle" input_arg { name: "value" diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt index 064b5c62bb..1b20981bb4 100644 --- a/tensorflow/core/ops/ops.pbtxt +++ b/tensorflow/core/ops/ops.pbtxt @@ -9307,6 +9307,55 @@ op { description: "`indices` is an K-dimensional integer tensor, best thought of as a\n(K-1)-dimensional tensor of indices into `params`, where each element defines a\nslice of `params`:\n\n output[i_0, ..., i_{K-2}] = params[indices[i0, ..., i_{K-2}]]\n\nWhereas in @{tf.gather} `indices` defines slices into the first\ndimension of `params`, in `tf.gather_nd`, `indices` defines slices into the\nfirst `N` dimensions of `params`, where `N = indices.shape[-1]`.\n\nThe last dimension of `indices` can be at most the rank of\n`params`:\n\n indices.shape[-1] <= params.rank\n\nThe last dimension of `indices` corresponds to elements\n(if `indices.shape[-1] == params.rank`) or slices\n(if `indices.shape[-1] < params.rank`) along dimension `indices.shape[-1]`\nof `params`. The output tensor has shape\n\n indices.shape[:-1] + params.shape[indices.shape[-1]:]\n\nSome examples below.\n\nSimple indexing into a matrix:\n\n```python\n indices = [[0, 0], [1, 1]]\n params = [[\'a\', \'b\'], [\'c\', \'d\']]\n output = [\'a\', \'d\']\n```\n\nSlice indexing into a matrix:\n\n```python\n indices = [[1], [0]]\n params = [[\'a\', \'b\'], [\'c\', \'d\']]\n output = [[\'c\', \'d\'], [\'a\', \'b\']]\n```\n\nIndexing into a 3-tensor:\n\n```python\n indices = [[1]]\n params = [[[\'a0\', \'b0\'], [\'c0\', \'d0\']],\n [[\'a1\', \'b1\'], [\'c1\', \'d1\']]]\n output = [[[\'a1\', \'b1\'], [\'c1\', \'d1\']]]\n\n\n indices = [[0, 1], [1, 0]]\n params = [[[\'a0\', \'b0\'], [\'c0\', \'d0\']],\n [[\'a1\', \'b1\'], [\'c1\', \'d1\']]]\n output = [[\'c0\', \'d0\'], [\'a1\', \'b1\']]\n\n\n indices = [[0, 0, 1], [1, 0, 1]]\n params = [[[\'a0\', \'b0\'], [\'c0\', \'d0\']],\n [[\'a1\', \'b1\'], [\'c1\', \'d1\']]]\n output = [\'b0\', \'b1\']\n```\n\nBatched indexing into a matrix:\n\n```python\n indices = [[[0, 0]], [[0, 1]]]\n params = [[\'a\', \'b\'], [\'c\', \'d\']]\n output = [[\'a\'], [\'b\']]\n```\n\nBatched slice indexing into a matrix:\n\n```python\n indices = [[[1]], [[0]]]\n params = [[\'a\', \'b\'], [\'c\', \'d\']]\n output = [[[\'c\', \'d\']], [[\'a\', \'b\']]]\n```\n\nBatched indexing into a 3-tensor:\n\n```python\n indices = [[[1]], [[0]]]\n params = [[[\'a0\', \'b0\'], [\'c0\', \'d0\']],\n [[\'a1\', \'b1\'], [\'c1\', \'d1\']]]\n output = [[[[\'a1\', \'b1\'], [\'c1\', \'d1\']]],\n [[[\'a0\', \'b0\'], [\'c0\', \'d0\']]]]\n\n indices = [[[0, 1], [1, 0]], [[0, 0], [1, 1]]]\n params = [[[\'a0\', \'b0\'], [\'c0\', \'d0\']],\n [[\'a1\', \'b1\'], [\'c1\', \'d1\']]]\n output = [[[\'c0\', \'d0\'], [\'a1\', \'b1\']],\n [[\'a0\', \'b0\'], [\'c1\', \'d1\']]]\n\n\n indices = [[[0, 0, 1], [1, 0, 1]], [[0, 1, 1], [1, 1, 0]]]\n params = [[[\'a0\', \'b0\'], [\'c0\', \'d0\']],\n [[\'a1\', \'b1\'], [\'c1\', \'d1\']]]\n output = [[\'b0\', \'b1\'], [\'d0\', \'c1\']]\n```" } op { + name: "GatherV2" + input_arg { + name: "params" + description: "The tensor from which to gather values. Must be at least rank\n`axis + 1`." + type_attr: "Tparams" + } + input_arg { + name: "indices" + description: "Index tensor. Must be in range `[0, params.shape[axis])`." + type_attr: "Tindices" + } + input_arg { + name: "axis" + description: "The axis in `params` to gather `indices` from. Defaults to the first\ndimension. Supports negative indexes." + type_attr: "Taxis" + } + output_arg { + name: "output" + description: "Values from `params` gathered from indices given by `indices`, with\nshape `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`." + type_attr: "Tparams" + } + attr { + name: "Tparams" + type: "type" + } + attr { + name: "Tindices" + type: "type" + allowed_values { + list { + type: DT_INT32 + type: DT_INT64 + } + } + } + attr { + name: "Taxis" + type: "type" + allowed_values { + list { + type: DT_INT32 + type: DT_INT64 + } + } + } + summary: "Gather slices from `params` axis `axis` according to `indices`." + description: "`indices` must be an integer tensor of any dimension (usually 0-D or 1-D).\nProduces an output tensor with shape `params.shape[:axis] + indices.shape +\nparams.shape[axis + 1:]` where:\n\n```python\n # Scalar indices (output is rank(params) - 1).\n output[a_0, ..., a_n, b_0, ..., b_n] =\n params[a_0, ..., a_n, indices, b_0, ..., b_n]\n\n # Vector indices (output is rank(params)).\n output[a_0, ..., a_n, i, b_0, ..., b_n] =\n params[a_0, ..., a_n, indices[i], b_0, ..., b_n]\n\n # Higher rank indices (output is rank(params) + rank(indices) - 1).\n output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =\n params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]\n```\n\n<div style=\"width:70%; margin:auto; margin-bottom:10px; margin-top:20px;\">\n<img style=\"width:100%\" src=\"https://www.tensorflow.org/images/Gather.png\" alt>\n</div>" +} +op { name: "GetSessionHandle" input_arg { name: "value" |