Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for TFE_NewContextOptions (0.18 sec)

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

    TEST(PARALLEL_DEVICE_LIB, TestOpWithError) {
      std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
          TF_NewStatus(), TF_DeleteStatus);
      std::unique_ptr<TFE_ContextOptions, decltype(&TFE_DeleteContextOptions)> opts(
          TFE_NewContextOptions(), TFE_DeleteContextOptions);
      std::unique_ptr<TF_Buffer, decltype(&TF_DeleteBuffer)> config(
          TF_CreateConfig(
              /*xla*/ false,
              /* gpu_memory_allow_growth */ true, /* num_cpu_devices */
    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_cluster_test.cc

                      server_def, tensorflow::Env::Default(), &worker_server)
                      .ok());
      ASSERT_TRUE(worker_server->Start().ok());
    
      TF_Status* status = TF_NewStatus();
      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);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_unified_experimental_test.cc

      }
    };
    
    TEST_P(UnifiedCAPI, TestBasicEager) {
      std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
          TF_NewStatus(), TF_DeleteStatus);
      TFE_ContextOptions* opts = TFE_NewContextOptions();
      TFE_ContextOptionsSetTfrt(opts, std::get<1>(GetParam()));
      TF_ExecutionContext* ctx = TF_NewEagerExecutionContext(opts, status.get());
      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 39.1K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_distributed_test.cc

                      server_def, tensorflow::Env::Default(), &worker_server2)
                      .ok());
      ASSERT_TRUE(worker_server2->Start().ok());
    
      TF_Status* status = TF_NewStatus();
      TFE_ContextOptions* opts = TFE_NewContextOptions();
      TFE_ContextOptionsSetAsync(opts, static_cast<unsigned char>(/*enable=*/true));
      TFE_ContextOptionsSetDevicePlacementPolicy(opts, TFE_DEVICE_PLACEMENT_SILENT);
      TFE_Context* ctx = TFE_NewContext(opts, 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)
  5. tensorflow/c/eager/c_api.cc

      tensorflow::AttrValue value;
      SetAttrValue("kEagerRuntime", &value);
      (*function_def.mutable_attr())["_construction_context"] = value;
    }
    
    }  // namespace
    
    extern "C" {
    
    TFE_ContextOptions* TFE_NewContextOptions() { return new TFE_ContextOptions; }
    
    void TFE_ContextOptionsSetConfig(TFE_ContextOptions* options, const void* proto,
                                     size_t proto_len, TF_Status* status) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top