- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for their (0.29 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.cc
// // TODO(allenl): There may be smarter ways to do this copy in some // cases, i.e. with a collective broadcast. We'll need to be careful // about things that are taken as inputs on the host or on their // existing device (for multi-device functions). std::unique_ptr<ParallelTensor> parallel_tensor( parallel_device.CopyToParallelDevice(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
if (dev->strict_scope_placement && *requested_placement == '\0') { 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;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
// CPU tensor to GPU). // Setting a larger thread pool does not help with the Swift caller, as we use // a different TFE context for each thread of execution (for running graph // functions, and their send/recvs corountines). config.set_inter_op_parallelism_threads(1); TF_Buffer* ret = TF_NewBuffer(); TF_CHECK_OK(MessageToBuffer(config, ret)); return ret; }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc
std::vector<char> x(block_size, 'x'); EXPECT_EQ(out, x); })); } // The `threads` destructor blocks until the threads can be joined, once their // respective reads finish (which happens once they are all concurrently being // executed, or 10 seconds have passed). } TEST(RamFileBlockCacheTest, CoalesceConcurrentReads) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:57 GMT 2021 - 23.2K bytes - Viewed (0) -
tensorflow/c/c_api_function_test.cc
ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); TF_GraphCopyFunction(host_graph_, func2, grad_func, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); // Verify that functions and their gradients are in host graph's GraphDef GraphDef gdef; GetGraphDef(host_graph_, &gdef); std::vector<std::pair<string, string>> grads = GetGradDefs(gdef); ASSERT_EQ(2, grads.size());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 20 22:08:54 GMT 2023 - 63.6K bytes - Viewed (6)