[tfjs-converter] add encodeBase64, decodeBase64

Author: vabarbosa

tfjs-converter/docs/supported_ops.md

@@ -266,6 +266,13 @@
 |Not mapped|ifft|
 |Not mapped|rfft|
 
+## Operations - Strings
+
+|Tensorflow Op Name|Tensorflow.js Op Name|
+|---|---|
+|DecodeBase64|decodeBase64|
+|EncodeBase64|encodeBase64|
+
 ## Tensors - Transformations
 
 |Tensorflow Op Name|Tensorflow.js Op Name|

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

@@ -0,0 +1,31 @@
+[
+  {
+    "tfOpName": "DecodeBase64",
+    "category": "string",
+    "inputs": [
+      {
+        "start": 0,
+        "name": "input",
+        "type": "tensor"
+      }
+    ]
+  },
+  {
+    "tfOpName": "EncodeBase64",
+    "category": "string",
+    "inputs": [
+      {
+        "start": 0,
+        "name": "input",
+        "type": "tensor"
+      }
+    ],
+    "attrs": [
+      {
+        "tfName": "pad",
+        "name": "pad",
+        "type": "bool"
+      }
+    ]
+  }
+]

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

@@ -0,0 +1,46 @@
+/**
+ * @license
+ * Copyright 2019 Google Inc. All Rights Reserved.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * =============================================================================
+ */
+
+import * as tfc from '@tensorflow/tfjs-core';
+
+import {NamedTensorsMap} from '../../data/types';
+import {ExecutionContext} from '../../executor/execution_context';
+import {InternalOpExecutor, Node} from '../types';
+
+import {getParamValue} from './utils';
+
+export let executeOp: InternalOpExecutor =
+    (node: Node, tensorMap: NamedTensorsMap,
+     context: ExecutionContext): tfc.Tensor[] => {
+      switch (node.op) {
+        case 'DecodeBase64': {
+          const input =
+              getParamValue('str', node, tensorMap, context) as tfc.Tensor;
+          return [tfc.decodeBase64(input)];
+        }
+        case 'EncodeBase64': {
+          const input =
+              getParamValue('str', node, tensorMap, context) as tfc.Tensor;
+          const pad = getParamValue('pad', node, tensorMap, context) as boolean;
+          return [tfc.encodeBase64(input, pad)];
+        }
+        default:
+          throw TypeError(`Node type ${node.op} is not implemented`);
+      }
+    };
+
+export const CATEGORY = 'string';

tfjs-converter/src/operations/op_list/string.ts

@@ -0,0 +1,32 @@
+import {OpMapper} from '../types';
+
+/**
+ * @license
+ * Copyright 2019 Google LLC. All Rights Reserved.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * =============================================================================
+ */
+
+export const json: OpMapper[] = [
+  {
+    'tfOpName': 'DecodeBase64',
+    'category': 'string',
+    'inputs': [{'start': 0, 'name': 'input', 'type': 'tensor'}]
+  },
+  {
+    'tfOpName': 'EncodeBase64',
+    'category': 'string',
+    'inputs': [{'start': 0, 'name': 'input', 'type': 'tensor'}],
+    'attrs': [{'tfName': 'pad', 'name': 'pad', 'type': 'bool'}]
+  }
+];

tfjs-converter/src/operations/types.ts

@@ -1,6 +1,6 @@
 /**
  * @license
- * Copyright 2018 Google LLC. All Rights Reserved.
+ * Copyright 2019 Google LLC. All Rights Reserved.
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
@@ -25,7 +25,7 @@ export type ParamType = 'number'|'string'|'string[]'|'number[]'|'bool'|'bool[]'|
 export type Category =
     'arithmetic'|'basic_math'|'control'|'convolution'|'custom'|'dynamic'|
     'evaluation'|'image'|'creation'|'graph'|'logical'|'matrices'|
-    'normalization'|'reduction'|'slice_join'|'spectral'|'transformation';
+    'normalization'|'reduction'|'slice_join'|'spectral'|'string'|'transformation';
 
 // For mapping input or attributes of NodeDef into TensorFlow.js op param.
 export declare interface ParamMapper {