Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetKernelSpec (0.16 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)
  3. tensorflow/compiler/mlir/lite/quantization/quantization_context.cc

    }
    
    LogicalResult QuantizeContext::Handle(
        quantfork::QuantizeRegionOp op,
        llvm::SmallVectorImpl<Operation *> *new_items, bool *changed) {
      auto signature = GetSignature(op);
      auto spec = target_spec_.GetKernelSpec(op.getLogicalKernel(), signature);
      if (!spec.has_value()) {
        op.emitWarning(
            "Couldn't find kernel from the registration for quantization.");
        return success();
      }
      switch (spec->type) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 08 01:38:03 UTC 2024
    - 13.1K bytes
    - Viewed (0)
Back to top