Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TFE_NewContext (0.07 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_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)
  3. 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)
  4. tensorflow/c/eager/c_api.cc

      options->device_placement_policy = policy;
    }
    
    void TFE_DeleteContextOptions(TFE_ContextOptions* options) { delete options; }
    
    TFE_Context* TFE_NewContext(const TFE_ContextOptions* opts, TF_Status* status) {
      if (opts->use_tfrt) {
        status->status = tensorflow::errors::Unimplemented("TFRT is not supported");
        return nullptr;
      }
    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