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