- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 178 for getDefiningOp (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/gpu_fusion.cc
batch_norm = dyn_cast_or_null<FusedBatchNormV3Op>(add_op.getX().getDefiningOp()); if (batch_norm) { side_input = add_op.getY(); } else { // Didn't get a FusedBatchNorm on the LHS of the AddV2, try the RHS. batch_norm = dyn_cast_or_null<FusedBatchNormV3Op>(add_op.getY().getDefiningOp()); if (!batch_norm) return failure(); side_input = add_op.getX(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 5.2K 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/tfr/passes/rewrite_quantized_io.cc
if (!returned_type || !returned_type.getElementType().isa<quant::QuantizedType>()) { continue; } if (auto returned_op = returned_value.get().getDefiningOp<TFR::CastOp>()) { auto new_type = returned_type.clone(returned_type.getElementType() .cast<quant::QuantizedType>()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize.cc
PatternRewriter &rewriter) { auto dot = dyn_cast_or_null<mhlo::DotGeneralOp>( reshape_after.getOperand().getDefiningOp()); if (!dot) return failure(); auto reshape_before = dyn_cast_or_null<mhlo::ReshapeOp>(dot.getLhs().getDefiningOp()); if (!reshape_before) return failure(); if (!dot.getLhs().getType().hasStaticShape() || !dot.getRhs().getType().hasStaticShape() ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 26.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_splits.cc
for (Value operand : op->getOperands()) { if (Operation* pred = operand.getDefiningOp()) { int result_number = llvm::cast<OpResult>(operand).getResultNumber(); if (auto pred_island = llvm::dyn_cast<tf_executor::IslandOp>(pred)) { Value yield_operand = pred_island.GetYield().getOperand(result_number); predecessors.push_back(yield_operand.getDefiningOp()); } } } return predecessors; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 18:44:34 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
Operation *op_input = args[0].getDefiningOp(); Operation *op_weight = args[1].getDefiningOp(); if (isa<TF::SubOp>(op_input)) { op_input = op_input->getOperand(0).getDefiningOp(); } if (isa<TF::SubOp>(op_weight)) { op_weight = op_weight->getOperand(0).getDefiningOp(); } if (isa<TF::CastOp>(op_input)) { op_input = op_input->getOperand(0).getDefiningOp();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc
if (!fetch.getDefiningOp()) { // Skip, because there is no op to add control to (eg: function args). continue; } else if (fetch.getDefiningOp()->getParentOp() == island_op) { // Skip, because it is the same island. continue; } else if (auto other_island_op = llvm::dyn_cast<tf_executor::IslandOp>( fetch.getDefiningOp())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
auto read_var = llvm::dyn_cast_or_null<TF::ReadVariableOp>( operand.get().getDefiningOp()); if (!read_var) continue; auto partitioned_input = llvm::dyn_cast_or_null<TF::TPUPartitionedInputV2Op>( read_var.getResource().getDefiningOp()); if (!partitioned_input || !AllResourceTypesHaveSubtypes(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 11.8K 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/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc
bool IsQintValueDefinedByIntToQintCast(Value v) { if (!IsIllegalType(v.getType())) { return true; } if (!v.getDefiningOp() || !llvm::isa<TF::CastOp>(v.getDefiningOp())) { return false; } return IsIntToQintCast(v.getDefiningOp()); } bool IsTFUniformQuantizedOpLegal(Operation *op) { // UniformQuantized Ops are considered legal if its qint operands and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.6K bytes - Viewed (0)