Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for tfe_context_options_ (0.12 sec)

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

    using ::testing::HasSubstr;
    
    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,
    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);
    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

        CHECK_EQ(errors::OK, s.code()) << s.message();
      }
    };
    
    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());
    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);
    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

    }
    
    }  // 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) {
      TF_SetConfig(&options->session_options, proto, proto_len, status);
    }
    
    void TFE_ContextOptionsSetAsync(TFE_ContextOptions* options,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_experimental.cc

      return static_cast<TFE_MonitoringSamplerCell*>(
          static_cast<void*>(sampler->sampler->GetCell(label1, label2)));
    }
    
    void TFE_ContextOptionsSetTfrt(TFE_ContextOptions* options, bool use_tfrt) {
      options->use_tfrt = use_tfrt;
    }
    
    TFE_CancellationManager* TFE_NewCancellationManager() {
      return tensorflow::wrap(new tensorflow::CancellationManager);
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 35.9K bytes
    - Viewed (0)
Back to top