- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 55 for getDefiningOp (0.3 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.cc
value_select.getOperand(0).getDefiningOp()); if (!value_gt || value_gt.getComparisonDirection() != compare_direction_included || value_gt.getLhs() != body.getArgument(0) || value_gt.getRhs() != body.getArgument(2)) return failure(); mhlo::SelectOp index_select = llvm::dyn_cast_or_null<mhlo::SelectOp>( return_op.getOperand(1).getDefiningOp()); if (!index_select) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/constant_fold.cc
SmallVector<ElementsAttr> inputs; for (auto operand : op->getOperands()) { auto preceding_const_op = operand.getDefiningOp<TF::ConstOp>(); if (preceding_const_op) { inputs.push_back(preceding_const_op.getValue()); continue; } Operation* preceding_op = operand.getDefiningOp(); int preceding_result_id = -1; for (auto preceding_result : preceding_op->getResults()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td
// See the function doc in the header file. def GetPadOpType : NativeCodeCall< "GetPadOpType((*$0.begin()).getDefiningOp<mhlo::CompositeOp>())">; // See the function doc in the header file. def GetAvgPoolOpPadAttr: NativeCodeCall<"GetAvgPoolOpPadAttr($_builder, (*$0.begin()).getDefiningOp<mhlo::CompositeOp>())">; // Returns true if the provided padding in the composite op can *not* be
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fuse_convolution_pass.cc
Value rhs = mul_op.getRhs(); conv_op = lhs.getDefiningOp<mhlo::ConvolutionOp>(); if (conv_op == nullptr) { return failure(); } filter = conv_op.getRhs().getDefiningOp<mhlo::ConstantOp>(); if (filter == nullptr) { return failure(); } // Try to match static broadcast or dynamic broadcast. bcast_or_const_op = rhs.getDefiningOp(); bool is_dynamic_broadcast =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 22:21:19 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/optimize.td
include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td" include "mhlo/IR/hlo_ops.td" include "stablehlo/dialect/ChloOps.td" def IsDefinedByConvOrDotGeneralOp : Constraint<Or<[ CPred<"$0.getDefiningOp<mhlo::ConvolutionOp>()">, CPred<"$0.getDefiningOp<mhlo::DotGeneralOp>()">]>>; def IsNull : Constraint<CPred<"!$0">>; // This pattern optimizes: // conv/dot_general + a + b -> conv/dot_general + (a + b)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 24 02:26:47 UTC 2024 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
bool NotFromDequant(mlir::Value value) { auto dequant_op = value.getDefiningOp<DequantizeOp>(); if (dequant_op) { return false; } auto split_op = value.getDefiningOp<SplitOp>(); if (!split_op) { return true; } return !split_op.getValue().getDefiningOp<DequantizeOp>(); } // Optimize TFLite operations in functions. class OptimizeBatchMatmulPass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.td
// `tf_quant.composite_function` attribute. def IsNotInLiftedFunc : Constraint<CPred<"!IsInLiftedFunc($0.getDefiningOp())">>; // Checks if the value is not inside a StableHLO op with region. def IsNotInStableHloOpRegion : Constraint<CPred<"!IsInStableHloOpRegion($0.getDefiningOp())">>; // Checks if the given einsum op is supported for XlaDotV2 quantization. def IsEinsumSupportedByXlaDotV2 :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 00:32:20 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.td
class AreIntSplatValuesEqual<string type> : Constraint<CPred< "AreSplatValuesEqual<"# type #">($0, $1)">>; // Returns defining op of this value. def GetDefiningOp : NativeCodeCall<"$0.getDefiningOp()">; // Clones an operation with new operands while keeping attributes. def CloneOpWithReplacedOperands : NativeCodeCall< "CloneOpWithReplacedOperands("
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 04:55:44 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
LogicalResult matchAndRewrite(TF::LookupTableFindV2Op find_op, PatternRewriter& rewriter) const override { auto handle_op = find_op.getTableHandle().getDefiningOp(); if (handle_op == nullptr) return failure(); auto hashtable_op = llvm::dyn_cast<TFL::HashtableOp>(handle_op); if (hashtable_op == nullptr) return failure(); rewriter.replaceOpWithNewOp<TFL::HashtableFindOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_device_copy_conversion.cc
// * the arg to the identity op is produced by a TPUExecuteOp. if (op_device && op_device.getValue().contains("TPU")) { return true; } Operation *def_op = arg.getDefiningOp(); // If the arg to this identity op is the arg of a function, there's no // defining op. if (def_op != nullptr && (isa<TF::TPUExecuteOp, TF::TPUExecuteAndUpdateVariablesOp>(def_op))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0)