Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for ParseCustomOpSpecs (0.25 sec)

  1. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_config.cc

    namespace quant {
    namespace {
    bool GetBooleanSpecs(const std::string& bool_val) {
      bool result;
      std::stringstream iss(bool_val);
      iss >> std::boolalpha >> result;
      return result;
    }
    }  // namespace
    
    void ParseCustomOpSpecs(const absl::string_view node_names,
                            const CustomOpUpdateOptions& update_option,
                            CustomOpMap& custom_op_map) {
      if (node_names.empty()) return;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 07:39:40 UTC 2024
    - 5.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/quantize.cc

        quant_specs.nodes_blocklist = absl::flat_hash_set<std::string>(
            nodes_blocklist_flag_.begin(), nodes_blocklist_flag_.end());
      }
    
      if (!enable_custom_op_weight_only_.empty()) {
        ParseCustomOpSpecs(enable_custom_op_weight_only_,
                           quant::CustomOpUpdateOptions::kWeightOnly,
                           quant_specs.custom_map);
      }
      if (enable_float16_quantization_) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 13.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_config.h

      // to be eliminated and fused into quantized kernels.
      QDQConversionMode qdq_conversion_mode = QDQConversionMode::kQDQNone;
    };
    
    // Parses the command line flag strings to the CustomOpMap specification.
    void ParseCustomOpSpecs(absl::string_view node_names,
                            const CustomOpUpdateOptions& update_option,
                            CustomOpMap& custom_op_map);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 13 10:16:19 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/post_quantize.cc

    };
    
    #include "tensorflow/compiler/mlir/lite/transforms/generated_post_quantize.inc"
    
    void PostQuantizePass::runOnOperation() {
      if (!enable_custom_op_no_side_effect_.empty()) {
        ParseCustomOpSpecs(enable_custom_op_no_side_effect_,
                           quant::CustomOpUpdateOptions::kNoSideEffect,
                           custom_op_map_);
      }
    
      RewritePatternSet patterns(&getContext());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_dynamic_range.cc

          !enable_dynamic_range_per_channel_quantization_for_dense_layers_;
      quant_specs_.minimum_elements_for_weights = min_elements_for_weights_;
    
      if (!enable_custom_op_quantization_.empty()) {
        ParseCustomOpSpecs(enable_custom_op_quantization_,
                           quant::CustomOpUpdateOptions::kInputIndices,
                           quant_specs_.custom_map);
      }
    
      ConvertTFLQuantOpsToMlirQuantOps(func);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20.8K bytes
    - Viewed (0)
Back to top