- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for SplitOp (0.09 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.h
}; // Unroll split into a bunch of slice ops. struct UnrollSplit : public OpRewritePattern<TFL::SplitOp> { using OpRewritePattern<TFL::SplitOp>::OpRewritePattern; LogicalResult matchAndRewrite(TFL::SplitOp split_op, PatternRewriter& rewriter) const override; }; // Unroll splitv into a bunch of slice ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 03 16:37:16 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_splits.cc
void runOnOperation() override; }; void TPUColocateSplits::runOnOperation() { getOperation().walk([&](Operation* op) { if (auto split = llvm::dyn_cast<TF::SplitOp>(op)) { if (HasDevice(split) || split->getAttrOfType<ArrayAttr>(kClassAttr)) return WalkResult::advance(); for (Operation* pred : IslandPredecessors(split)) { if (auto colocation_classes =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 18:44:34 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
bool NotFromDequant(mlir::Value value) { auto dequant_op = value.getDefiningOp<DequantizeOp>(); if (dequant_op) { return false; } auto split_op = value.getDefiningOp<SplitOp>(); if (!split_op) { return true; } return !split_op.getValue().getDefiningOp<DequantizeOp>(); } // Optimize TFLite operations in functions. class OptimizeBatchMatmulPass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0)