save

Author: nkreeger

binding/tfjs_backend.cc

@@ -19,6 +19,7 @@
 #include "tfe_auto_op.h"
 #include "tfe_utils.h"
 #include "utils.h"
+#include "../deps/tensorflow/include/tensorflow/c/c_api.h"
 
 namespace tfnodejs {
 
@@ -251,4 +252,34 @@ napi_value TFJSBackend::ExecuteOp(napi_env env, napi_value op_name_value,
   return output_tensor_infos;
 }
 
+napi_value TFJSBackend::ExecuteOpFastPath(napi_env env,
+                                          napi_value op_name_value,
+                                          napi_value input_tensor_ids,
+                                          napi_value num_output_values) {
+  napi_status nstatus;
+  TF_AutoStatus tf_status;
+
+  char op_name[NAPI_STRING_SIZE];
+  nstatus = napi_get_value_string_utf8(env, op_name_value, op_name,
+                                       NAPI_STRING_SIZE, nullptr);
+  ENSURE_NAPI_OK_RETVAL(env, nstatus, nullptr);
+
+  // Crack open and look at the op registry.
+  TF_Buffer* op_list_buffer = TF_GetAllOpList();
+  TF_ApiDefMap* api_def_map = TF_NewApiDefMap(op_list_buffer, tf_status.status);
+  ENSURE_TF_OK_RETVAL(env, tf_status, nullptr);
+
+  // // tensorflow::ApiDef api_def;
+  // tensorflow::ApiDef api_def;
+  // api_def.ParseFromArray(api_def_buffer->data, api_def_buffer->length);
+  // fprintf(stderr, "apidef.name: %s\n", api_def.graph_op_name.c_str());
+
+  // For a test - don't return anything.
+  napi_value output_tensor_infos;
+  nstatus = napi_create_array_with_length(env, 0, &output_tensor_infos);
+  ENSURE_NAPI_OK_RETVAL(env, nstatus, nullptr);
+
+  return output_tensor_infos;
+}
+
 }  // namespace tfnodejs

binding/tfjs_backend.h

@@ -58,6 +58,11 @@ class TFJSBackend {
                        napi_value op_attr_inputs, napi_value input_tensor_ids,
                        napi_value num_output_values);
 
+  // TODO(kreeger): Document me.
+  napi_value ExecuteOpFastPath(napi_env env, napi_value op_name_value,
+                               napi_value input_tensor_ids,
+                               napi_value num_output_values);
+
  private:
   int32_t InsertHandle(TFE_TensorHandle* tfe_handle);
 

binding/tfjs_binding.cc

@@ -123,6 +123,25 @@ static napi_value ExecuteOp(napi_env env, napi_callback_info info) {
   return gBackend->ExecuteOp(env, args[0], args[1], args[2], args[3]);
 }
 
+static napi_value ExecuteOpFastPath(napi_env env, napi_callback_info info) {
+  napi_status nstatus;
+
+  size_t argc = 3;
+  napi_value args[argc];
+  napi_value js_this;
+  nstatus = napi_get_cb_info(env, info, &argc, args, &js_this, nullptr);
+  ENSURE_NAPI_OK_RETVAL(env, nstatus, nullptr);
+
+  if (argc < 3) {
+    NAPI_THROW_ERROR(env, "Invalid number of args passed to executeOp()");
+    return nullptr;
+  }
+
+  ENSURE_VALUE_IS_STRING_RETVAL(env, args[0], nullptr);
+  ENSURE_VALUE_IS_ARRAY_RETVAL(env, args[1], nullptr);
+  ENSURE_VALUE_IS_NUMBER_RETVAL(env, args[2], nullptr);
+}
+
 static napi_value InitTFNodeJSBinding(napi_env env, napi_value exports) {
   napi_status nstatus;