- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 176 for getDefiningOp (0.17 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_restore_op.cc
Value assigned_value_operand = assign_variable_op.getOperand(1); if (auto var_handle_op = dyn_cast<TF::VarHandleOp>(resource_operand.getDefiningOp()); var_handle_op && isa<TF::ConstOp>(assigned_value_operand.getDefiningOp())) { var_handle_ops.emplace_back(var_handle_op); } } return var_handle_ops; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Mar 12 06:02:20 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
Value pad_input = pad_op.getOperand(); RankedTensorType pad_type = pad_op.getType().cast<RankedTensorType>(); auto transpose_op = pad_input.getDefiningOp<stablehlo::TransposeOp>(); if (!transpose_op || !transpose_op->hasOneUse()) return failure(); Value transpose_input = transpose_op.getOperand(); ArrayRef<int64_t> transpose_perm = transpose_op.getPermutation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/fold_constant_transpose.cc
LogicalResult match(mlir::stablehlo::TransposeOp op) const override { Value operand = op.getOperand(); auto const_op = dyn_cast_or_null<mlir::stablehlo::ConstantOp>(operand.getDefiningOp()); if (!const_op) return failure(); // Only support float tensors. auto tensor_type = mlir::dyn_cast_or_null<TensorType>(const_op.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.td
def FuseAffineOpAndMul : Pat< (TF_MulOp (SupportedAffineOpMatcher $conv_out, $input, $weight), (TF_ConstOp:$mul_rhs IsFloatElementsAttr:$mul_rhs_value)), (CloneOpWithReplacedOperands (GetDefiningOp $conv_out), $input, (MultiplyFakeQuantValue $weight, (MakeOneDimValueBroadcastable $mul_rhs, $weight))), [(HasOneUse $conv_out), (HasRankOf<1> $mul_rhs_value),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 03:24:59 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
auto var_handle_op = dyn_cast<TF::VarHandleOp>(resource_operand.getDefiningOp()); if (!var_handle_op) continue; Value assigned_value_operand = assign_variable_op.getOperand(1); auto const_op = dyn_cast<TF::ConstOp>(assigned_value_operand.getDefiningOp()); if (!const_op) continue; var_handle_ops.emplace_back(var_handle_op); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dilated_conv.h
} // Check if the ConvOp's input is defined by `Expand` op, and the output used // by `Squeeze` op. Operation* producer_op = op.getOperand(0).getDefiningOp(); if (!producer_op || producer_op->getNumResults() != 1) { return rewriter.notifyMatchFailure( op, "op doesn't have a producer node that has a single result"); } if (!producer_op->hasOneUse() ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op_test.cc
quant_spec.set_tensor_type(QuantizationComponentSpec::TENSORTYPE_INT_8); std::optional<TF::PartitionedCallOp> dequantize_op = ApplyUniformQuantization( pattern_rewriter, cast<TF::ConstOp>(value.getDefiningOp()), quant_spec); EXPECT_TRUE(dequantize_op.has_value()); EXPECT_EQ(dequantize_op.value().func().getName().str(), "composite_dequantize_uniform"); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc
ShapedType result_type, StringRef func_name, Value& func_input_arg) { Operation* input_op = input_val.getDefiningOp(); Operation* insertion_point = input_op->getParentOfType<func::FuncOp>(); if (!insertion_point) insertion_point = input_op->getParentOfType<ModuleOp>(); rewriter.setInsertionPointAfter(insertion_point);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0)