Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for enable_stablehlo_conversion (0.3 sec)

  1. tensorflow/compiler/mlir/lite/common/tfl_pass_config.h

                << pass_config.guarantee_all_funcs_one_use
                << "\nenable_hlo_to_tf_conversion: "
                << pass_config.enable_hlo_to_tf_conversion
                << "\nenable_stablehlo_conversion: "
                << pass_config.enable_stablehlo_conversion
                << "\nlegalize_custom_tensor_list_ops: "
                << pass_config.legalize_custom_tensor_list_ops
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 19:05:30 UTC 2024
    - 6.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/python/graphdef_to_tfl_flatbuffer.cc

      pass_config.preserve_assert_op = toco_flags.preserve_assert_op();
      pass_config.guarantee_all_funcs_one_use =
          toco_flags.guarantee_all_funcs_one_use();
      pass_config.enable_stablehlo_conversion = toco_flags.convert_to_stablehlo();
    
      // StableHLO Quantizer is not supported for GraphDef inputs, so
      // quantization_py_function_lib is set to nullptr.
      return internal::ConvertMLIRToTFLiteFlatBuffer(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 11 19:29:56 UTC 2024
    - 5.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.h

    extern llvm::cl::opt<bool> disable_hlo_to_tfl_conversion;
    
    // quantization related flags
    extern llvm::cl::opt<bool> post_training_quantization;
    
    // TF to stablehlo pass flags
    extern llvm::cl::opt<bool> enable_stablehlo_conversion;
    
    // Wether serialize stablehlo ops or not
    extern llvm::cl::opt<bool> serialize_stablehlo_ops;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 20:53:17 UTC 2024
    - 3.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/python/jax_to_tfl_flatbuffer.cc

        pass_config.unfold_batch_matmul = false;
      }
      pass_config.unfold_large_splat_constant =
          toco_flags.unfold_large_splat_constant();
      pass_config.enable_hlo_to_tf_conversion = true;
      pass_config.enable_stablehlo_conversion = toco_flags.convert_to_stablehlo();
    
      mlir::OwningOpRef<mlir::ModuleOp> module;
      if (model_flags.hlo_file_type() == toco::ModelFlags::HLO_TEXT) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 11 19:29:56 UTC 2024
    - 8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.cc

    // NOLINTNEXTLINE
    opt<bool> preserve_assert_op(
        "preserve-assert-op",
        llvm::cl::desc("Preserve AssertOp during tfl legalization."),
        llvm::cl::init(false));
    
    // NOLINTNEXTLINE
    opt<bool> enable_stablehlo_conversion(
        "enable-stablehlo-conversion",
        llvm::cl::desc("Enable converting TF to Stablehlo."),
        llvm::cl::init(false));
    
    // NOLINTNEXTLINE
    opt<bool> post_training_quantization(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 20:53:17 UTC 2024
    - 7.9K bytes
    - Viewed (0)
Back to top