Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for RegisterXlaDeviceKernels (0.26 sec)

  1. tensorflow/compiler/jit/xla_cpu_device.cc

      registration.cluster_slow_ops = true;
      registration.cluster_inaccurate_ops = true;
      XlaOpRegistry::RegisterCompilationDevice(DEVICE_XLA_CPU, registration);
    
      static XlaDeviceOpRegistrations* registrations =
          RegisterXlaDeviceKernels(DEVICE_XLA_CPU, DEVICE_CPU_XLA_JIT);
      (void)registrations;
    
      TF_ASSIGN_OR_RETURN(auto platform,
                          se::PlatformManager::PlatformWithName("Host"));
    
      XlaDevice::Options options;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 5.5K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. tensorflow/compiler/jit/xla_gpu_device.cc

      registration.cluster_inaccurate_ops = true;
      XlaOpRegistry::RegisterCompilationDevice(DEVICE_XLA_GPU, registration);
    
      static XlaDeviceOpRegistrations* registrations =
          RegisterXlaDeviceKernels(DEVICE_XLA_GPU, DEVICE_GPU_XLA_JIT);
      (void)registrations;
    
      auto platform = se::PlatformManager::PlatformWithName(se::GpuPlatformName());
      if (!platform.ok()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 6.6K bytes
    - Viewed (0)
  5. 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