Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TFE_OpGetContext (0.22 sec)

  1. tensorflow/c/eager/c_api.cc

        return;
      }
    
      tensorflow::unwrap(op)->Release();
    }
    
    const char* TFE_OpGetName(const TFE_Op* op, TF_Status* status) {
      return tensorflow::unwrap(op)->Name().c_str();
    }
    
    TFE_Context* TFE_OpGetContext(const TFE_Op* op, TF_Status* status) {
      return tensorflow::wrap(tensorflow::unwrap(op)->GetContext());
    }
    
    void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_test.cc

    // Needs to work with a const TFE_Op since custom devices should not modify the
    // op they are called with.
    TFE_Op* CloneOp(const TFE_Op* other) {
      TF_Status* status = TF_NewStatus();
      TFE_Context* context = TFE_OpGetContext(other, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      const char* op_name = TFE_OpGetName(other, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
Back to top