- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 1,179 for OP (0.03 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td
//===----------------------------------------------------------------------===// // DivNoNan and MulNonNan op patterns. //===----------------------------------------------------------------------===// class BinaryNoNanPat<Op FromOp, Op ToOp> : Pat<(FromOp $l, $r), (TF_SelectV2Op (TF_EqualOp $r, (TF_ConstOp:$zero (GetScalarOfType<0> $r)),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
auto loader, tensorflow::XlaCallModuleLoader::Create( context, static_cast<int>(op.getVersion()), op.getModule().str(), std::move(disabled_checks), std::move(platforms), /*num_invocation_args=*/op.getArgs().size(), op.getHasTokenInputOutput())); return std::move(*loader).module(); } // Renames functions in the stablehlo module to avoid naming conflicts with
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_weights.cc
// And the dequant_op op will propagate towards quant_op only. if (failed(quantizeOps(rewriter, op, weight_component_spec.value()))) { return failure(); } return success(); } private: // Check if op's user or op's user after an identity op is connected to a // terminator. bool checkIfAnyUserIsConnectedToTermiantor(BlockArgument op) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc
InitializeArgState(arg, value); } fn_.walk([&](Operation* op) { std::unique_ptr<OpQuantScaleSpec> scale_spec = GetQuantScaleSpec(op); if (!IsOpQuantizable(op) && !scale_spec->has_same_scale_requirement) { return; } work_list_.push_back(op); for (int i = 0; i < op->getNumOperands(); ++i) { Value operand = op->getOperand(i); if (Operation* inst = operand.getDefiningOp()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 38.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/verify_quant_legalization.cc
IsTFUniformQuantizedOp(op) || IsMhloUniformQuantizedOp(*op)) { op->emitOpError("is illegal as it is a UQ op or contains uq/qint types"); LOG(ERROR) << "Found illegal op containing uq/qint type: " << op->getName().getStringRef().str(); return WalkResult::interrupt(); } return WalkResult::advance(); }); if (walk_result.wasInterrupted()) { signalPassFailure(); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_type_precision.cc
dyn_cast<arith::ConstantOp>(op.getOperand(0).getDefiningOp()); if (!input_op) { return failure(); } Builder builder(op.getContext()); auto new_gather_op = rewriter.create<TFL::GatherOp>( op.getLoc(), /*result=*/ mlir::cast<TensorType>(op.getResult().getType()) .clone(builder.getI4Type()), /*operand=*/op.getOperands(), op->getAttrs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_weight_param.cc
LogicalResult match(Operation* op) const override { if (op->getNumResults() != 1) { return failure(); } auto type = mlir::cast<TensorType>(op->getResult(0).getType()); if (!type || !type.getElementType().isF32()) { return failure(); } return success( op->hasOneUse() && IsWeightQuantizableFunction(*op->getUses().begin(), type.getRank())); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.h
// Runs the analysis and populates `control_predecessors_` for `op`. void AnalyzeOp(Operation* op); // Updates `control_predecessors_` for given `resource_id` and `op`. void AddPredecessorsForAccess(ResourceId resource_id, Operation* op, bool read_only); // Updates resource access for given `resource_id` and `op` in // `per_resource_access_info_` and `op_to_resource_ids_`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.h
// Returns the entry function name for the given tf.XlaCallModule op. Returns // empty string if such attribute does not exist. StringRef GetEntryFunctionName(TF::XlaCallModuleOp op); // Checks whether the given op contains QuantizationTrait::FullyQuantizable. inline bool HasQuantizableTrait(Operation* op) { return op->hasAttrOfType<StringAttr>(kQuantTraitAttrName) && op->getAttrOfType<StringAttr>(kQuantTraitAttrName).getValue().str() ==
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/constant_fold.cc
SmallVector<Value> ConstantFoldOpIfPossible(Operation* op) { if (!IsOperationFoldable(op)) return op->getResults(); OpBuilder builder(op); SmallVector<Value> results; if (failed(FoldOperation(builder, op, results))) { return op->getResults(); } return results; } LogicalResult ConstantFoldQuantizableOperands::matchAndRewrite( Operation* op, PatternRewriter& rewriter) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 5K bytes - Viewed (0)