Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for FindFirstOpFromMainFunc (0.3 sec)

  1. 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)
  2. 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)
Back to top