Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 12 for TFE_Executor (0.21 sec)

  1. tensorflow/c/eager/tfe_executor_internal.h

    ==============================================================================*/
    #ifndef TENSORFLOW_C_EAGER_TFE_EXECUTOR_INTERNAL_H_
    #define TENSORFLOW_C_EAGER_TFE_EXECUTOR_INTERNAL_H_
    
    #include <memory>
    
    #include "tensorflow/core/common_runtime/eager/eager_executor.h"
    
    struct TFE_Executor {
      explicit TFE_Executor(bool async, bool enable_streaming_enqueue,
                            int in_flight_nodes_limit)
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Nov 22 20:30:59 GMT 2022
    - 1.5K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_experimental_test.cc

      ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status);
      TF_DeleteFunction(fn);
    
      for (bool async : {false, true, false}) {
        TFE_Executor* old_executor = TFE_ContextGetExecutorForThread(ctx);
        TFE_Executor* executor = TFE_NewExecutor(
            /*is_async=*/async, /*enable_streaming_enqueue=*/true,
            /*in_flight_nodes_limit=*/0);
        TFE_ContextSetExecutorForThread(ctx, executor);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 03:14:26 GMT 2023
    - 31.5K bytes
    - Viewed (1)
  3. tensorflow/c/eager/c_api_experimental.cc

    }
    
    TFE_Executor* TFE_NewExecutor(bool is_async, bool enable_streaming_enqueue,
                                  int in_flight_nodes_limit) {
      return new TFE_Executor(is_async, enable_streaming_enqueue,
                              in_flight_nodes_limit);
    }
    
    void TFE_DeleteExecutor(TFE_Executor* executor) { delete executor; }
    
    bool TFE_ExecutorIsAsync(TFE_Executor* executor) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
  4. tensorflow/c/eager/c_api_experimental.h

        TFE_Op* op, TFE_CancellationManager* cancellation_manager,
        TF_Status* status);
    
    // -----------------------------------------------------------------------------
    // Eager Executor APIs.
    typedef struct TFE_Executor TFE_Executor;
    
    // Creates a new eager Executor. Nodes in one executor are guaranteed to be
    // executed in sequence. Assigning nodes to different executors allows executing
    // nodes in parallel.
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_internal.h

    #include "tensorflow/c/eager/c_api.h"
    #include "tensorflow/c/eager/c_api_experimental.h"
    #include "tensorflow/c/eager/tfe_cancellation_manager_internal.h"  // IWYU pragma: export
    #include "tensorflow/c/eager/tfe_executor_internal.h"  // IWYU pragma: export
    #include "tensorflow/c/eager/tfe_monitoring_internal.h"  // IWYU pragma: export
    #include "tensorflow/c/eager/tfe_op_attrs_internal.h"  // IWYU pragma: export
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Jan 18 19:26:34 GMT 2023
    - 1.9K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_remote_test.cc

      TFE_DeleteTensorHandle(h1_task0);
      TFE_DeleteTensorHandle(h0_task1);
      TFE_DeleteTensorHandle(h1_task1);
      TFE_DeleteTensorHandle(retvals[0]);
    
      TFE_DeleteOp(matmul);
    
      TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx);
      TFE_ExecutorWaitForAllPendingNodes(executor, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_DeleteExecutor(executor);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Aug 12 00:14:22 GMT 2020
    - 5.4K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_cluster_test.cc

      CheckTFE_TensorHandleHasFloats(retval_task0, {7, 10, 15, 22});
    
      TFE_DeleteTensorHandle(retval_task0);
      TFE_DeleteTensorHandle(h0_task0);
      TFE_DeleteTensorHandle(retvals[0]);
    
      TFE_DeleteOp(matmul);
    
      TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx);
      TFE_ExecutorWaitForAllPendingNodes(executor, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_DeleteExecutor(executor);
      TF_DeleteStatus(status);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Apr 14 10:03:59 GMT 2023
    - 19.3K bytes
    - Viewed (0)
  8. tensorflow/c/eager/BUILD

            "//tensorflow:internal",
        ],
        deps = [
            "//tensorflow/c:conversion_macros",
            "//tensorflow/core:framework",
        ],
    )
    
    cc_library(
        name = "tfe_executor_internal",
        hdrs = ["tfe_executor_internal.h"],
        visibility = ["//visibility:private"],
        deps = [
            "//tensorflow/core/common_runtime/eager:eager_executor",
        ],
    )
    
    cc_library(
    Plain Text
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 33.3K bytes
    - Viewed (0)
  9. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

    };
    
    using StatusPtr = std::unique_ptr<TF_Status, StatusDeleter>;
    
    class ExecutorDeleter {
     public:
      void operator()(TFE_Executor* to_delete) const {
        TFE_DeleteExecutor(to_delete);
      }
    };
    
    using ExecutorPtr = std::unique_ptr<TFE_Executor, ExecutorDeleter>;
    
    }  // namespace
    
    // Allows a single op at a time to be launched without blocking.
    //
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  10. tensorflow/c/eager/c_api_remote_test_util.cc

      for (auto* h : handles_task0) {
        TFE_DeleteTensorHandle(h);
      }
      for (auto* h : handles_task2) {
        TFE_DeleteTensorHandle(h);
      }
    
      TFE_DeleteOp(matmul);
    
      TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx);
      TFE_ExecutorWaitForAllPendingNodes(executor, status);
      ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
      TFE_DeleteExecutor(executor);
      if (func) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Dec 11 22:56:03 GMT 2020
    - 9.1K bytes
    - Viewed (0)
Back to top