- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for BroadcastFoldPass (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/fold_broadcast.cc
}; #define GEN_PASS_DEF_BROADCASTFOLDPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" class BroadcastFoldPass : public impl::BroadcastFoldPassBase<BroadcastFoldPass> { public: void runOnOperation() override; }; LogicalResult ConvertResultsBroadcastableShapeOp::matchAndRewrite( Operation* op, PatternRewriter& rewriter) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
"clEnumValN(MoveTransposeDirection::kEnd, \"end\", \"end of the block\"))"> ]; } def BroadcastFoldPass : Pass<"tf-broadcast-fold", "mlir::func::FuncOp"> { let summary = "Fold explicit broadcasts into the following operations if they " "support implicit broadcasting on their operand.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)