Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 122 for eraseOp (0.23 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top