- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 22 for target_ops_ (0.17 sec)
-
tensorflow/compiler/mlir/lite/transforms/raise_custom_ops.cc
this->target_ops_ = target_ops; } void runOnOperation() override; }; void RaiseCustomOpsPass::runOnOperation() { auto fn = getOperation(); OpBuilder builder(fn.getContext()); absl::flat_hash_set<std::string> target_op_names(target_ops_.begin(), target_ops_.end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/passes.td
let summary = "Raise custom ops into tflite dialect."; let constructor = "CreateRaiseCustomOpsPass()"; let dependentDialects = ["TFL::TensorFlowLiteDialect"]; let options = [ ListOption<"target_ops_", "test-raise-tf-targets", "std::string", "comma separated list of target op names to be wrapped. Only used in tests">, ]; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 22.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions_drq.cc
// Constructor used by manually creating the pass. explicit LiftQuantizableSpotsAsFunctionsDRQPass( const QuantMethod quantization_method, const OpSet target_opset, const int min_num_elements_for_weights) { quantization_method_ = quantization_method; target_opset_ = target_opset; min_num_elements_for_weights_ = min_num_elements_for_weights; } LiftQuantizableSpotsAsFunctionsDRQPass(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc
OpSet target_opset) : quant_specs_(quant_specs) { weight_quantization_ = quant_specs.weight_quantization; target_opset_ = target_opset; } QuantizePass(const QuantizePass& other) : quant_specs_(other.quant_specs_) { weight_quantization_ = other.weight_quantization_; target_opset_ = other.target_opset_; } StringRef getArgument() const final {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
quantization_method_ = quantization_method; target_opset_ = target_opset; enable_per_channel_quantization_ = enable_per_channel_quantization; } QuantizeCompositeFunctionsPass(const QuantizeCompositeFunctionsPass& other) { quantization_method_ = other.quantization_method_; target_opset_ = other.target_opset_; enable_per_channel_quantization_ = other.enable_per_channel_quantization_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
'has_bias': [True, False], 'has_batch_norm': [True, False], 'target_opset': [quant_opts_pb2.XLA], }]) ) @test_util.run_in_graph_and_eager_modes def test_qat_conv_model( self, activation_fn: Optional[ops.Operation], has_bias: bool, has_batch_norm: bool, target_opset: quant_opts_pb2.OpSet, ): class ConvModel(module.Module):
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/check_control_dependencies.cc
} } }); // Find all control sources for `target_op` for which we should not have a // dependency and emit corresponding warnings. for (const auto& [target_op, control_sources] : op_to_control_sources) { if (IsIntermediateOp(target_op)) continue; for (IslandOp source_op : control_sources) { if (!ShouldOpsHaveDependency(source_op, target_op, analysis_for_func)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 10.2K bytes - Viewed (0) -
common/scripts/setup_env.sh
exit 1 fi LOCAL_OS=$(uname) # Pass environment set target operating-system to build system if [[ ${TARGET_OS} ]]; then # Target explicitly set : elif [[ $LOCAL_OS == Linux ]]; then TARGET_OS=linux readlink_flags="-f" elif [[ $LOCAL_OS == Darwin ]]; then TARGET_OS=darwin readlink_flags="" else echo "This system's OS, $LOCAL_OS, isn't supported" exit 1 fi
Registered: Fri Jun 14 15:00:06 UTC 2024 - Last Modified: Wed Jun 12 14:37:27 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
IntegerAttr num_bits; BoolAttr narrow_range; ::llvm::SmallVector<Operation *, 4> target_ops; target_ops.push_back(op); ReorderOp old_reorder_op = ::llvm::dyn_cast<ReorderOp>(op); if (failed(findTargetOps(old_reorder_op, rewriter, max, min, input, num_bits, narrow_range, target_ops))) { return failure(); } shape = old_reorder_op.getODSOperands(1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/passes.h
// Apply graph optimizations such as fusing and constant folding to prepare // lifting. std::unique_ptr<OperationPass<func::FuncOp>> CreatePrepareLiftingPass( tensorflow::quantization::OpSet target_opset); // Lifts the dynamic range quantizable spots as composite functions. std::unique_ptr<OperationPass<ModuleOp>> CreateLiftQuantizableSpotsAsFunctionsDRQPass( tensorflow::quantization::QuantizationMethod::PresetMethod
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 12.3K bytes - Viewed (0)