- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 234 for eraseOp (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
to_remove.push_back(init_func_op); } else { to_keep.push_back(sym_ref); } } for (auto func_op : to_remove) rewriter.eraseOp(func_op); if (to_keep.empty()) rewriter.eraseOp(op); else op->setAttr("initializers", rewriter.getArrayAttr(to_keep)); return success(); } }; void SessionInitializerOp::getCanonicalizationPatterns(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_tf_control_flow_to_scf.cc
TypeRange block_arguments_type, PatternRewriter& rewriter) { // If `destination_block` isn't empty, erase its terminator to ensure that it // never contains two terminator-like ops after merging. if (!destination_block->empty()) rewriter.eraseOp(destination_block->getTerminator()); destination_block->addArguments( block_arguments_type,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
if ((custom_op_map.find(op_name) == custom_op_map.end()) || !custom_op_map.find(op_name)->second.no_side_effect) return failure(); } rewriter.eraseOp(op); return success(); } quant::CustomOpMap custom_op_map; }; #include "tensorflow/compiler/mlir/lite/transforms/generated_post_quantize.inc" void PostQuantizePass::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
rewriter.replaceAllUsesWith(assign_group.getResult(0), const_group_size); rewriter.replaceAllUsesWith(assign_group.getResult(1), const_group_key); rewriter.eraseOp(assign_group); return success(); } }; void LegalizeTFCollective::runOnOperation() { // FIXME(b/226139061): Figure out a way to share the channel_id with
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc
return failure(); for (auto &use : llvm::make_early_inc_range(op.getControl().getUses())) use.getOwner()->eraseOperand(use.getOperandNumber()); rewriter.eraseOp(op); return success(); } }; // This pattern matches and removes IslandOps with no inner ops, no control // operands, one data result and no control result user. The single data result
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
return failure(); } // TF Lite doesn't support Assert, we just drop the assert from the graph. LogicalResult ConvertTFAssertOp::matchAndRewrite( Operation* op, PatternRewriter& rewriter) const { rewriter.eraseOp(op); return success(); } // Legalize unidirectional sequence lstm. struct LegalizeUnidirectionalSequenceLstm : public RewritePattern { explicit LegalizeUnidirectionalSequenceLstm(MLIRContext* context)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
for (Operation *user : llvm::make_early_inc_range(old_result.getUsers())) rewriter.replaceOp(user, ValueRange(new_result)); } // Erase original unpack operation. rewriter.eraseOp(op.getOperation()); return success(); } } // namespace void UnpackOp::getCanonicalizationPatterns(RewritePatternSet &results,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
PatternRewriter& rewriter) const override { ElementsAttr cst; if (matchPattern(op.getCondition(), m_Constant(&cst))) { if (cst.getValues<bool>()[0]) { rewriter.eraseOp(op); return success(); } } return failure(); } }; } // namespace void AssertOp::getCanonicalizationPatterns(RewritePatternSet& results,
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
conv_op.getLoc(), new_conv.getResult(), rewriter.getI64TensorAttr(transpose_order)); conv_op->replaceAllUsesWith(output_transpose); rewriter.eraseOp(conv_op); } else { SmallVector<int32_t, 4> output_shape_i32; for (int64_t dim : mlir::cast<RankedTensorType>(conv_op.getResult().getType()) .getShape()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
values_slice_op.erase(); } if (!indices.use_empty()) { auto indices_slice_op = llvm::dyn_cast_or_null<TFL::SliceOp>( indices.getUses().begin().getUser()); indices_slice_op.getResult().replaceAllUsesWith(top_k_op.getIndices()); indices_slice_op.erase(); } op.erase(); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0)