- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for FindFirstOpFromMainFunc (0.19 sec)
-
tensorflow/compiler/mlir/quantization/common/test_base.h
// representation of a `ModuleOp` containing a single function `@main`. // Returns `failure` iff there is no `@main` or no such operation is found in // `@main`. template <typename OpT> FailureOr<OpT> FindFirstOpFromMainFunc(ModuleOp module_op) { func::FuncOp main_func_op = FindMainFuncOp(module_op); if (main_func_op == nullptr) return failure(); auto ops = main_func_op.getOps<OpT>(); if (ops.empty()) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec_test.cc
const OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kXlaCallModuleOpWithPerTensorQuantizedConvolution); ASSERT_TRUE(module_op); const FailureOr<TF::XlaCallModuleOp> xla_call_module_op = FindFirstOpFromMainFunc<TF::XlaCallModuleOp>(*module_op); ASSERT_TRUE(succeeded(xla_call_module_op)); const std::unique_ptr<OpQuantSpec> op_quant_spec = GetStableHloOpQuantSpec(*xla_call_module_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
const OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kXlaCallModuleOpWithQuantizationMethodAttr); ASSERT_TRUE(module_op); FailureOr<TF::XlaCallModuleOp> xla_call_module_op = FindFirstOpFromMainFunc<TF::XlaCallModuleOp>(*module_op); ASSERT_TRUE(succeeded(xla_call_module_op)); // Test that `GetQuantizationMethodOrDefault` returns a valid `Method` // corresponding to `"no_quantization {}"`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0)