Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TFE_ContextClearCaches (0.24 sec)

  1. tensorflow/c/eager/c_api.h

                                                                TF_Status* status);
    
    // Clears the internal caches in the TFE context. Useful when reseeding random
    // ops.
    TF_CAPI_EXPORT extern void TFE_ContextClearCaches(TFE_Context* ctx);
    
    // Sets a thread-local device placement policy. After this call, other calls to
    // TFE_Execute in the same thread will use the device policy specified here
    Registered: Tue Oct 29 12:39:09 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

    }
    
    TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) {
      TF_DeviceList* l = new TF_DeviceList;
      tensorflow::unwrap(ctx)->ListDevices(&l->response);
      return l;
    }
    
    void TFE_ContextClearCaches(TFE_Context* ctx) {
      tensorflow::unwrap(ctx)->ClearCachesAndThreadExecutors();
    }
    
    // Set server_def on the context, possibly updating it.
    TF_CAPI_EXPORT extern void TFE_ContextSetServerDef(TFE_Context* ctx,
    Registered: Tue Oct 29 12:39:09 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_test.cc

                                status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      for (bool clear_cache : {true, false, true}) {
        if (clear_cache) {
          TFE_ContextClearCaches(ctx);
        }
        TFE_TensorHandle* m = TestMatrixTensorHandle(ctx);
        TFE_TensorHandle* retval[1] = {nullptr};
        int num_retvals = 1;
        TFE_Op* op = TFE_NewOp(ctx, "MatMulFunction", status);
    Registered: Tue Oct 29 12:39:09 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
Back to top