- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for asynchronous (0.15 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
auto allocator = std::make_unique<AsyncValueAllocator>(); tensorflow::Tensor tensor(allocator.get(), DT_FLOAT, {1}); TF_ASSERT_OK_AND_ASSIGN( auto pjrt_client, xla::GetTfrtCpuClient(/*asynchronous=*/true, /*cpu_device_count=*/1)); std::vector<int32_t> data(1, 0); xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1}); TF_ASSERT_OK_AND_ASSIGN( auto buffer, pjrt_client->BufferFromHostBuffer(
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
ExpectScalarEq<float>(result_components[1].get(), 3.); } TEST(PARALLEL_DEVICE, TestCollectiveSync) { TestCollective(/*async=*/false); } // Note that ops on the parallel device currently don't execute // asynchronously. The test is just that we don't get deadlocks. TEST(PARALLEL_DEVICE, TestCollectiveAsync) { TestCollective(/*async=*/true); } void RegisterCollectiveMulFunction(TFE_Context* context,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1)