- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getNumRegions (0.33 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
auto new_op = CreateTfOp<CaseOrIfRegionOp>(rewriter, op, new_result_types, op.getOperand(), op->getAttrs(), op.getNumRegions()); int next_index = 0; for (auto result : op.getResults()) { if (!result_to_extern_value.count(result)) { result.replaceAllUsesWith(new_op.getResult(next_index++)); continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
LogicalResult matchAndRewrite( mhlo::WhileOp while_op, OpAdaptor adaptor, ConversionPatternRewriter& rewriter) const final { // HLO WhileOp should have two regions: cond and body. if (while_op->getNumRegions() != 2) return failure(); // This rule doesn't support mhlo::WhileOp with tuple inputs. for (auto type : while_op->getOperandTypes()) { if (mlir::isa<TupleType>(type)) return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)