- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for expectClose (0.12 sec)
-
tensorflow/compiler/jit/xla_host_send_recv_device_context_test.cc
new XlaHostRecvDeviceContext(stream.get(), gpu_dst, shape, done_event); TF_ASSERT_OK(device_context->CopyDeviceTensorToCPUSync( &device_tensor, "", device_.get(), &dest_cpu_tensor)); tensorflow::test::ExpectClose(origin_cpu_tensor, dest_cpu_tensor); device_context->Unref(); } TEST_F(XlaHostSendRecvDeviceContextTest, CopyCPUTensorToDevice) { SetDevice("GPU");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 7.2K bytes - Viewed (0)