- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TFE_OpReset (0.67 sec)
-
tensorflow/c/eager/custom_device_test.cc
TFE_OpReset(reused_op.get(), "Identity", custom_device_name, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); ASSERT_EQ(tensorflow::string(TFE_OpGetDevice(reused_op.get(), status.get())), tensorflow::string(custom_device_name)); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TFE_OpReset(reused_op.get(), "Identity",
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
if (TF_GetCode(status) != TF_OK) return; TFE_OpSetDevice(op_.get(), device_.c_str(), status); if (TF_GetCode(status) != TF_OK) return; } else { TFE_OpReset(op_.get(), operation_name, device_.c_str(), status); if (TF_GetCode(status) != TF_OK) return; } TFE_OpAddAttrs(op_.get(), attributes); for (int input_index = 0; input_index < inputs.size(); ++input_index) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental.cc
#include "tensorflow/core/platform/errors.h" #include "tensorflow/core/platform/mutex.h" #include "tensorflow/core/platform/strcat.h" #include "tsl/framework/cancellation.h" using tensorflow::string; void TFE_OpReset(TFE_Op* op_to_reset, const char* op_or_function_name, const char* raw_device_name, TF_Status* status) { if (op_to_reset) { tensorflow::ImmediateExecutionOperation* op =
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/eager/c_api_experimental.h
// than separately calling it because if the existing op has the same // `raw_device_name`, it skips parsing and just leave as it is. TF_CAPI_EXPORT extern void TFE_OpReset(TFE_Op* op_to_reset, const char* op_or_function_name, const char* raw_device_name, TF_Status* status);
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
TFE_Op* matmul = TFE_NewOp(ctx, "MatMul", status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_TensorHandle* retvals[1]; int num_retvals = 1; for (auto s : state) { TFE_OpReset(matmul, "MatMul", nullptr, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddInput(matmul, m, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1)