- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for RegisterXlaDeviceKernels (0.36 sec)
-
tensorflow/compiler/jit/xla_device.cc
// already non-ok, so there's nothing extra to report if HandleDeviceError // itself returns an error. HandleDeviceError().IgnoreError(); } return status; } XlaDeviceOpRegistrations* RegisterXlaDeviceKernels( const char* device, const char* jit_device, OpKernel* (*factory)(OpKernelConstruction*), StringPiece kernel_class_name) { XlaOpRegistry::RegisterCompilationKernels();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.h
op_kernel_registrars; }; XlaDeviceOpRegistrations* RegisterXlaDeviceKernels( const char* device, const char* jit_device, OpKernel* (*factory)(OpKernelConstruction*), StringPiece kernel_class_name); XlaDeviceOpRegistrations* RegisterXlaDeviceKernels(const char* device, const char* jit_device);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
registration.cluster_inaccurate_ops = true; XlaOpRegistry::RegisterCompilationDevice(DEVICE_TPU_NODE, registration); static XlaDeviceOpRegistrations* registrations = RegisterXlaDeviceKernels(DEVICE_TPU_NODE, DEVICE_TPU_XLA_JIT); (void)registrations; int device_count = platform->VisibleDeviceCount(); VLOG(1) << "Creating " << device_count << " TPU devices";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0)