Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 4 of 4 for TFE_OpGetContext (0.17 seconds)

  1. tensorflow/c/eager/custom_device_testutil.cc

        TF_SetStatus(s, TF_INTERNAL,
                     "Ops must be placed on the device explicitly, or their inputs "
                     "first copied to other devices.");
        return;
      }
      TFE_Context* context = TFE_OpGetContext(original_op, s);
      if (TF_GetCode(s) != TF_OK) return;
      const char* operation_name = TFE_OpGetName(original_op, s);
      if (TF_GetCode(s) != TF_OK) return;
      const TFE_OpAttrs* attributes = TFE_OpGetAttrs(original_op);
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Dec 20 23:30:21 GMT 2025
    - 8.4K bytes
    - Click Count (0)
  2. tensorflow/c/eager/c_api.h

    TF_CAPI_EXPORT extern const char* TFE_OpGetName(const TFE_Op* op,
                                                    TF_Status* status);
    TF_CAPI_EXPORT extern TFE_Context* TFE_OpGetContext(const TFE_Op* op,
                                                        TF_Status* status);
    
    TF_CAPI_EXPORT extern void TFE_OpSetDevice(TFE_Op* op, const char* device_name,
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Click Count (0)
  3. 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) {
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Fri Nov 07 05:55:21 GMT 2025
    - 43.9K bytes
    - Click Count (0)
  4. 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);
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Oct 09 05:56:18 GMT 2025
    - 94.6K bytes
    - Click Count (0)
Back to Top