- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for OpTy (0.05 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_canonicalization_helper.h
// another TF op. // TODO(jpienaar): This is a workaround until behavior is established. template <typename OpTy, typename... Args> OpTy CreateTfOp(RewriterBase &b, Operation *op, Args &&...args) { auto ret = b.create<OpTy>(op->getLoc(), std::forward<Args>(args)...); CopyDeviceAndUnderscoredAttributes(op, ret.getOperation()); return ret; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Nov 12 21:57:12 UTC 2021 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.h
#include "mlir/Support/LLVM.h" // from @llvm-project #include "tfrt/basic_kernels/opdefs/types.h" // from @tf_runtime namespace tfrt { namespace fallback_common { template <typename OpTy> mlir::LogicalResult VerifyExecuteOpCommon(OpTy op) { auto op_attr_array = op.getOpAttrs().getValue(); for (auto op_attr : op_attr_array) { auto key_value = mlir::dyn_cast<mlir::ArrayAttr>(op_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/smuggle_disallowed_ops.cc
rewriter.replaceOp(op, custom_call.getResults()); return success(); } } // namespace template <typename OpTy> class SmuggleOpPattern : public OpRewritePattern<OpTy> { public: using OpRewritePattern<OpTy>::OpRewritePattern; LogicalResult matchAndRewrite(OpTy op, PatternRewriter& rewriter) const override { return SmuggleOp(op, rewriter); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 01:14:04 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
// output. template <typename OpTy> struct PruneUnusedOpsWithSideEffect : public OpRewritePattern<OpTy> { public: explicit PruneUnusedOpsWithSideEffect( MLIRContext* context, const quant::CustomOpMap& custom_op_map = {}) : OpRewritePattern<OpTy>(context), custom_op_map(custom_op_map) {} LogicalResult matchAndRewrite(OpTy op,
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.cc
// Derived: is the resulting derived class of this class. // OpTy: is TF::ArgMaxOp or TF::ArgMinOp. template <typename Derived, typename OpTy> class ConvertArgMinMaxOp : public OpRewritePattern<OpTy> { using OpRewritePattern<OpTy>::OpRewritePattern; LogicalResult matchAndRewrite(OpTy op, PatternRewriter &rewriter) const override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.h
}; // Clustering policy for a specific operation type. template <typename OpTy> class OpClusteringPolicy : public ClusteringPolicy { public: LogicalResult MatchAndUpdateConstraints( Operation* operation, const ValuesConstraintSet& results, ValuesConstraintSet& operands) const final { if (auto op = dyn_cast<OpTy>(operation)) return MatchAndUpdateConstraints(op, results, operands);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 03:47:00 UTC 2023 - 12.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
// // TODO(jpienaar): Evaluate benefit of templating here. template <typename OpTy, bool use_log = true> class LowerSoftmaxOp : public OpRewritePattern<OpTy> { public: using OpRewritePattern<OpTy>::OpRewritePattern; LogicalResult matchAndRewrite(OpTy op, PatternRewriter &rewriter) const override { Value logits = op.getLogits();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// 32-bit mask attributes. // - None of the strides values are zero. // - Ellipsis mask can have at most one bit set. template <class OpTy> static LogicalResult VerifyStridedSliceBase(OpTy op) { // Expected size for operands begin, end and strides vector operands. int64_t expected_size = -1; for (Value val : {op.getBegin(), op.getEnd(), op.getStrides()}) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)