- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for TensorShapeToXLAShape (0.23 sec)
-
tensorflow/compiler/jit/xla_host_send_recv_device_context_test.cc
TF_ASSERT_OK_AND_ASSIGN(auto stream, executor->CreateStream()); se::DeviceMemoryBase gpu_dst{device_tensor.data(), 4 * sizeof(float)}; xla::Shape shape; TF_ASSERT_OK(TensorShapeToXLAShape(DT_FLOAT, TensorShape({2, 2}), &shape)); // Copy the cpu_tensor to the GPU first before trying to copy it back. TF_ASSERT_OK( stream->Memcpy(&gpu_dst, origin_cpu_tensor.data(), gpu_dst.size()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 7.2K bytes - Viewed (0)