- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for RegisterCustomDevice (0.22 sec)
-
tensorflow/c/eager/immediate_execution_context.h
// already registered. // TODO(tfrt-devs): Remove this method. Let caller register it directly into // CustomDeviceOpHandler. virtual Status RegisterCustomDevice(const string& name, std::unique_ptr<CustomDevice> device) = 0; // Return FunctionLibraryDefinition. Transformations need to use it to use it // to invoke MLIR compiler passes.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 06 08:34:00 GMT 2023 - 12.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.h
// custom device. // // Many devices will want to simply return an "unimplemented" status // here. This is the default behavior if `pack` is null when passed to // TFE_RegisterCustomDevice. TFE_TensorHandle* (*pack)(TFE_Context* context, TFE_TensorHandle** handles, int num_handles, TF_Status* s, void* device_info) = nullptr;
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) -
tensorflow/c/eager/c_api.cc
} } // namespace extern "C" { bool TFE_IsCustomDevice(TFE_Context* ctx, const char* device_name) { return tensorflow::unwrap(ctx)->IsCustomDevice(device_name); } void TFE_RegisterCustomDevice(TFE_Context* ctx, TFE_CustomDevice device, const char* device_name, void* device_info, TF_Status* status) {
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)