- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for cancelFn (0.36 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
EXPECT_EQ(TF_GetCode(status.get()), TF_INVALID_ARGUMENT); EXPECT_THAT(TF_Message(status.get()), HasSubstr("assertion failed")); // Note that future collectives with the same context do not work at the // moment; once canceled, the collective executor requires the program to be // restarted / context to be reset. } TEST(PARALLEL_DEVICE_LIB, TestDifferentShapes) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
// // `cancellation_manager` must live until after `Join` finishes and pending // `is_async` operations finish. In addition to allowing the caller to cancel // the operation, its `StartCancel` method will be called if op execution // fails on any device in order to cancel the others. void StartExecute(TFE_Context* context, const char* operation_name, std::vector<TFE_TensorHandle*> inputs,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/c_api_test.cc
TEST(CAPI, Status) { TF_Status* s = TF_NewStatus(); EXPECT_EQ(TF_OK, TF_GetCode(s)); EXPECT_EQ(string(), TF_Message(s)); TF_SetStatus(s, TF_CANCELLED, "cancel"); EXPECT_EQ(TF_CANCELLED, TF_GetCode(s)); EXPECT_EQ(string("cancel"), TF_Message(s)); TF_DeleteStatus(s); } void Deallocator(void* data, size_t, void* arg) { tensorflow::cpu_allocator()->DeallocateRaw(data);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3)