- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CopyTensorHandleToDevice (0.13 sec)
-
tensorflow/c/eager/immediate_execution_context.h
virtual ImmediateExecutionTensorHandle* CreateLocalHandle( AbstractTensorInterface* t) = 0; // Copy the handle to another device. virtual ImmediateExecutionTensorHandle* CopyTensorHandleToDevice( ImmediateExecutionTensorHandle* handle, const char* device_name, absl::Status* status) = 0; // Create an operation to perform op execution
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
return nullptr; } tensorflow::ImmediateExecutionContext* unwrapped_ctx = tensorflow::unwrap(ctx); auto* result = unwrapped_ctx->GetCustomDeviceOpHandler().CopyTensorHandleToDevice( unwrapped_ctx, tensorflow::unwrap(h), device_name, &status->status); if (status->status.ok()) { return tensorflow::wrap(result); } return nullptr; }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0)