- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 122 for eraseOp (0.23 sec)
-
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) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
func_op.eraseArguments(args_to_erase_bit_vector); // Erases entries in "tf._input_shapes" attribute of `func_op` that correspond // to the erased arguments. if (auto input_shapes_attr = func_op->getAttrOfType<ArrayAttr>(kTfInputShapesAttr); input_shapes_attr) { // Construct a new array of input shapes excluding the input shapes of the // erased arguments. SmallVector<Attribute> updated_input_shapes_attr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
"of indices"); } } if ((indices_rank == 0) || (indices_rank == batch_dims_i)) { // Scalar indices (output is rank(params) - 1). // Erase shape[axis] shape.erase(shape.begin() + axis_i); } else if (indices_rank == 1) { // Vector indices (output is rank(params)). // Copy indices.shape into params.shape[axis]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
for (auto it : llvm::zip(control_node_op.getOutputs(), cloned_inner->getResults())) { std::get<0>(it).replaceAllUsesWith(std::get<1>(it)); } rewriter.eraseOp(outer_op); } return control_edges; } } // namespace namespace tflite { bool MlirToFlatBufferTranslateFunction(mlir::ModuleOp module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
} } return result; } void EraseReturnOperands(Region* region, llvm::BitVector& erase) { for (Block& block : region->getBlocks()) { for (Operation& op : block.getOperations()) { if (!op.hasTrait<OpTrait::ReturnLike>()) continue; op.eraseOperands(erase); } } } // Erases the given results from an operation, similar to what // Operation::eraseArguments does (but for results).
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0)