- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CustomDevice (0.09 sec)
-
tensorflow/c/eager/immediate_execution_context.h
#include "tensorflow/core/platform/tstring.h" #include "tensorflow/core/protobuf/config.pb.h" #include "tensorflow/core/util/device_name_utils.h" namespace tensorflow { class EagerExecutor; class EagerContext; class CustomDevice; class CustomDeviceOpHandler; class Device; // LINT.IfChange // Note: Keep in sync with exported copy of enum in eager/c_api.h. enum ContextDevicePlacementPolicy {
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; } return const_cast<void*>( static_cast<const void*>(tensor->tensor_data().data())); } namespace tensorflow { namespace { class CustomDeviceAPI : public tensorflow::CustomDevice { public: CustomDeviceAPI(TFE_Context* context, TFE_CustomDevice device, void* info, string name) : context_(context), device_(device), info_(info), name_(name) {}
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0)