- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetKernelSpec (0.16 sec)
-
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) -
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) -
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)