- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for RegisterCustomDevice (0.21 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.h
// the same shape, although this requirement may be relaxed in the future. // // `device_name` must not name an existing physical or custom device (see // the documentation for TFE_RegisterCustomDevice for more information). // // Tensors may be copied on or off the device explicitly using // TPUReplicatedInput and TPUReplicatedOutput respectively. For example, with
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jun 04 21:49:16 GMT 2020 - 2.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.h
TFE_CustomDevice device; void* device_info; tensorflow::parallel_device::AllocateParallelDevice( device_name, underlying_devices.data(), underlying_devices.size(), &device, &device_info); TFE_RegisterCustomDevice(context, device, device_name, device_info, status); } } // namespace parallel_device } // namespace tensorflow
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
device->executed_flag = executed_flag; device->device_name = name; device->underlying_device = "/job:localhost/replica:0/task:0/device:CPU:0"; device->strict_scope_placement = strict_scope_placement; TFE_RegisterCustomDevice(context, custom_device, name, device, status); } TFE_TensorHandle* UnpackTensorHandle(TFE_TensorHandle* logged_tensor_handle, TF_Status* status) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
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 23 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 23 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 23 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2)