Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 17 for TFE_NewContext (0.12 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

          TF_DeleteBuffer);
      TFE_ContextOptionsSetConfig(opts.get(), config->data, config->length,
                                  status.get());
      std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> context(
          TFE_NewContext(opts.get(), status.get()), TFE_DeleteContext);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
    
      std::vector<std::string> devices{
          "/job:localhost/replica:0/task:0/device:CPU:0",
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 15.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_test.cc

    using tensorflow::string;
    
    namespace {
    
    void BM_InitOp(::testing::benchmark::State& state) {
      TF_Status* status = TF_NewStatus();
      TFE_ContextOptions* opts = TFE_NewContextOptions();
      TFE_Context* ctx = TFE_NewContext(opts, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_DeleteContextOptions(opts);
    
      TFE_TensorHandle* m = TestMatrixTensorHandle(ctx);
      for (auto s : state) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_debug_test.cc

    #include "tensorflow/core/platform/test.h"
    
    TEST(CApiDebug, ScalarCPU) {
      TF_Status* status = TF_NewStatus();
      TFE_ContextOptions* opts = TFE_NewContextOptions();
      TFE_Context* ctx = TFE_NewContext(opts, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_DeleteContextOptions(opts);
    
      TFE_TensorHandle* h = TestScalarTensorHandle(ctx, 1.0f);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Apr 06 22:10:09 UTC 2020
    - 2.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/custom_device_test.cc

    TEST(CUSTOM_DEVICE, RegisterSimpleDevice) {
      std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
          TF_NewStatus(), TF_DeleteStatus);
      TFE_ContextOptions* opts = TFE_NewContextOptions();
      TFE_Context* context = TFE_NewContext(opts, status.get());
      TFE_DeleteContextOptions(opts);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      bool arrived = false;
      bool executed = false;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 27 23:39:24 UTC 2020
    - 18.4K bytes
    - Viewed (0)
  5. tensorflow/c/eager/parallel_device/parallel_device_test.cc

          TF_DeleteBuffer);
      TFE_ContextOptionsSetConfig(opts.get(), config->data, config->length,
                                  status.get());
      std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> context(
          TFE_NewContext(opts.get(), status.get()), TFE_DeleteContext);
      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
      BasicTestsForTwoDevices(context.get(),
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_experimental_test.cc

    }
    
    TEST(CAPI, ExecutorContextDestructionOrder) {
      TF_Status* status = TF_NewStatus();
    
      {
        TFE_ContextOptions* opts = TFE_NewContextOptions();
        TFE_Context* ctx = TFE_NewContext(opts, status);
        ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status);
        TFE_DeleteContextOptions(opts);
        TFE_Executor* executor = TFE_NewExecutor(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 03:14:26 UTC 2023
    - 31.5K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_cluster_test.cc

      TFE_ContextOptions* opts = TFE_NewContextOptions();
      TFE_ContextOptionsSetAsync(opts, static_cast<unsigned char>(async));
      TFE_ContextOptionsSetDevicePlacementPolicy(opts, TFE_DEVICE_PLACEMENT_SILENT);
      TFE_Context* ctx = TFE_NewContext(opts, status);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_DeleteContextOptions(opts);
    
      TFE_ContextSetServerDef(ctx, 0, serialized.data(), serialized.size(), status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  8. tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc

      std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
          TF_NewStatus(), TF_DeleteStatus);
      std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> context(
          TFE_NewContext(opts.get(), status.get()), TFE_DeleteContext);
      tensorflow::ServerDef server_def = GetServerDef("worker", 3);
    
      // This server def has the task index set to 0.
      std::string serialized = server_def.SerializeAsString();
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Jul 10 07:18:05 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api_distributed_test.cc

      TFE_ContextOptionsSetAsync(opts, static_cast<unsigned char>(/*enable=*/true));
      TFE_ContextOptionsSetDevicePlacementPolicy(opts, TFE_DEVICE_PLACEMENT_SILENT);
      TFE_Context* ctx = TFE_NewContext(opts, status);
      EXPECT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
      TFE_DeleteContextOptions(opts);
    
      TFE_ContextSetServerDef(ctx, 0, serialized.data(), serialized.size(), status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 23.4K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api_unified_experimental_eager.cc

    using tensorflow::wrap;
    using tensorflow::strings::StrCat;
    
    TF_ExecutionContext* TF_NewEagerExecutionContext(TFE_ContextOptions* options,
                                                     TF_Status* s) {
      TFE_Context* c_ctx = TFE_NewContext(options, s);
      if (TF_GetCode(s) != TF_OK) {
        return nullptr;
      }
      return wrap(static_cast<AbstractContext*>(unwrap(c_ctx)));
    }
    
    TF_AbstractTensor* TF_CreateAbstractTensorFromEagerTensor(TFE_TensorHandle* t,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Jun 25 04:40:46 UTC 2020
    - 3.2K bytes
    - Viewed (0)
Back to top