- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TFE_OpGetContext (0.13 sec)
-
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);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Mar 03 20:47:31 UTC 2021 - 8.3K bytes - Viewed (0) -
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,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
"first un-packed. Got an un-placed op with an input placed on the " "parallel device."); return; } TFE_Context* context = TFE_OpGetContext(original_op, status); if (TF_GetCode(status) != TF_OK) return; const char* operation_name = TFE_OpGetName(original_op, status); if (TF_GetCode(status) != TF_OK) return;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 18.3K bytes - Viewed (0) -
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: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0) -
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: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0)