Prechádzať zdrojové kódy

Update tf-metadata.json

Lutz Roeder 7 rokov pred
rodič
commit
cea4557bfc
2 zmenil súbory, kde vykonal 76 pridanie a 1 odobranie
  1. 76 0
      src/tf-metadata.json
  2. 0 1
      tools/tf

+ 76 - 0
src/tf-metadata.json

@@ -55582,6 +55582,35 @@
       "summary": "Get the number of nodes in a tree"
     }
   },
+  {
+    "name": "TensorListConcat",
+    "schema": {
+      "attributes": [
+        {
+          "name": "element_dtype",
+          "type": "type"
+        }
+      ],
+      "description": "Requires that all tensors have the same shape except the first dimension.\n\ninput_handle: The input list.\ntensor: The concated result.\nlengths: Output tensor containing sizes of the 0th dimension of tensors in the list, used for computing the gradient.\n",
+      "inputs": [
+        {
+          "name": "input_handle",
+          "type": 21
+        }
+      ],
+      "outputs": [
+        {
+          "name": "tensor",
+          "typeAttr": "element_dtype"
+        },
+        {
+          "name": "lengths",
+          "type": 9
+        }
+      ],
+      "summary": "Concats all tensors in the list along the 0th dimension."
+    }
+  },
   {
     "name": "TensorListConcatLists",
     "schema": {
@@ -55970,6 +55999,53 @@
       "summary": "Sets the index-th position of the list to contain the given tensor."
     }
   },
+  {
+    "name": "TensorListSplit",
+    "schema": {
+      "attributes": [
+        {
+          "name": "element_dtype",
+          "type": "type"
+        },
+        {
+          "allowedValues": [
+            {
+              "type": "type",
+              "value": 3
+            },
+            {
+              "type": "type",
+              "value": 9
+            }
+          ],
+          "name": "shape_type",
+          "type": "type"
+        }
+      ],
+      "description": "list[i] corresponds to lengths[i] tensors from the input tensor.\nThe tensor must have rank at least 1 and contain exactly sum(lengths) elements.\n\ntensor: The input tensor.\nelement_shape: A shape compatible with that of elements in the tensor.\nlengths: Vector of sizes of the 0th dimension of tensors in the list.\noutput_handle: The list.",
+      "inputs": [
+        {
+          "name": "tensor",
+          "typeAttr": "element_dtype"
+        },
+        {
+          "name": "element_shape",
+          "typeAttr": "shape_type"
+        },
+        {
+          "name": "lengths",
+          "type": 9
+        }
+      ],
+      "outputs": [
+        {
+          "name": "output_handle",
+          "type": 21
+        }
+      ],
+      "summary": "Splits a tensor into a list."
+    }
+  },
   {
     "name": "TensorListStack",
     "schema": {

+ 0 - 1
tools/tf

@@ -102,4 +102,3 @@ while [ "$#" != 0 ]; do
         "metadata") metadata;;
     esac
 done
-