Add TensorListLength op

tfjs-converter/docs/supported_ops.md

@@ -100,6 +100,7 @@
 |TensorListFromTensor|TensorListFromTensor|
 |TensorListGather|TensorListGather|
 |TensorListGetItem|TensorListGetItem|
+|TensorListLength|TensorListLength|
 |TensorListPopBack|TensorListPopBack|
 |TensorListPushBack|TensorListPushBack|
 |TensorListReserve|TensorListReserve|

tfjs-converter/python/tensorflowjs/op_list/control.json

@@ -810,5 +810,16 @@
         "type": "dtype"
       }
     ]
+  },
+  {
+    "tfOpName": "TensorListLength",
+    "category": "control",
+    "inputs": [
+      {
+        "start": 0,
+        "name": "tensorListId",
+        "type": "tensor"
+      }
+    ]
   }
-]
+]

tfjs-converter/src/metadata_test.ts

@@ -60,6 +60,7 @@ describe('kernel2op metadata file', () => {
       'TensorListFromTensor',
       'TensorListGather',
       'TensorListGetItem',
+      'TensorListLength',
       'TensorListPopBack',
       'TensorListPushBack',
       'TensorListReserve',

tfjs-converter/src/operations/executors/control_executor.ts

@@ -366,6 +366,12 @@ export const executeOp: InternalOpAsyncExecutor = async(
       context.addTensorList(tensorList);
       return [tensorList.idTensor];
     }
+    case 'TensorListLength': {
+      const idTensor =
+          getParamValue('tensorListId', node, tensorMap, context) as Tensor;
+      const tensorList = context.getTensorList(idTensor.id);
+      return [scalar(tensorList.size(), 'int32')];
+    }
     default:
       throw TypeError(`Node type ${node.op} is not implemented`);
   }

tfjs-converter/src/operations/executors/control_executor_test.ts

@@ -993,5 +993,29 @@ describe('control', () => {
         expect(validateParam(node, control.json)).toBeTruthy();
       });
     });
+
+    describe('TensorListLength', () => {
+      it('should get the size of tensorList', async () => {
+        const input4 = tensor1d([0, 0, 0], 'int32');
+        const input5 = tensor1d([1, 1, 1], 'int32');
+        const tensorList = new TensorList([input4, input5], [3], 'int32', 5);
+        context.addTensorList(tensorList);
+        node.op = 'TensorListLength';
+        node.inputParams['tensorListId'] = createTensorAttr(0);
+        node.inputNames = ['input2'];
+        const input2 = [tensorList.idTensor];
+        const size = await executeOp(node, {input2}, context);
+        expect(size.length).toEqual(1);
+        expect(size[0].shape).toEqual([]);
+        test_util.expectArraysClose(size[0].dataSync(), [2]);
+      });
+
+      it('should match json def', () => {
+        node.op = 'TensorListLength';
+        node.inputParams['tensorListId'] = createTensorAttr(0);
+
+        expect(validateParam(node, control.json)).toBeTruthy();
+      });
+    });
   });
 });