- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CustomDevice (0.19 sec)
-
tensorflow/c/eager/c_api.cc
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) {}
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/eager/c_api_experimental.h
// tensors/ops/etc. and usable in APIs like OpSetDevice/ResetOp/etc. #define TFE_CUSTOM_DEVICE_VERSION 4 // Struct to be filled in. Functions are required except where indicated. typedef struct TFE_CustomDevice { int version = TFE_CUSTOM_DEVICE_VERSION; // Method to copy a tensor to the custom device. TFE_TensorHandle* (*copy_tensor_to_device)(TFE_Context* context,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0)