- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for SetDevice (0.25 sec)
-
tensorflow/c/kernels_test.cc
TF_DeleteStatus(status); #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM std::unique_ptr<Device> device( DeviceFactory::NewDevice(device_name_, {}, "/job:a/replica:0/task:0")); OpsTestBase::SetDevice(DEVICE_GPU, std::move(device)); #endif TF_ASSERT_OK(NodeDefBuilder(op_name, op_name).Finalize(node_def())); TF_ASSERT_OK(InitOp()); } #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0)