Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for RegisterXlaDeviceKernels (0.36 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top