Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetKernelSpec (0.41 sec)

  1. tensorflow/compiler/mlir/lite/quantization/device_target.cc

      qi8n_ = AnyQuantizedType::get(kSigned, i8_, f32_, i8_min_ + 1, i8_max_);
      qi32_ = AnyQuantizedType::get(kSigned, i32_, f32_, i32_min_, i32_max_);
      assert(qi8n_ == qi8n_);
    }
    
    std::optional<KernelSpec> DeviceTarget::GetKernelSpec(
        llvm::StringRef kernel, const KernelSpecs::Signature& signature) const {
      auto kernel_specs_it = specs_.find(kernel);
      if (kernel_specs_it == specs_.end()) return std::nullopt;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 08 10:41:08 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/quantization/device_target.h

      ScaleDecomposeFn decompose_fn_;
    };
    
    class DeviceTarget {
     public:
      explicit DeviceTarget(MLIRContext* ctx);
    
      // Retrieves the kernel spec for the quant region op.
      std::optional<KernelSpec> GetKernelSpec(
          llvm::StringRef kernel, const KernelSpecs::Signature& signature) const;
    
      // Retrieves the scale decomposition function for the quant region op.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 08 10:41:08 UTC 2024
    - 7.1K bytes
    - Viewed (0)
Back to top