- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for set_nic_compatible (0.16 sec)
-
tensorflow/c/kernels_experimental.cc
for (int64_t i = 0; i < elements_in.size(); ++i) { elements_out(i) = elements_in(i); } } else { AllocatorAttributes attr; attr.set_gpu_compatible(true); attr.set_nic_compatible(true); TF_RETURN_IF_ERROR(context->allocate_temp( var->tensor()->dtype(), var->tensor()->shape(), &tmp, attr)); tensorflow::Status s; TF_Tensor* tf_tmp = TF_TensorFromTensor(tmp, &s);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:12:29 UTC 2024 - 30.9K bytes - Viewed (0)