- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for DropWhileShapeInvariantInDeviceClusterPass (0.72 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/drop_while_shape_invariant.cc
#define GEN_PASS_DEF_DROPWHILESHAPEINVARIANTINDEVICECLUSTERPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" class DropWhileShapeInvariantInDeviceClusterPass : public impl::DropWhileShapeInvariantInDeviceClusterPassBase< DropWhileShapeInvariantInDeviceClusterPass> { void runOnOperation() override; }; void DropWhileShapeInvariantAttr(Operation* op) { if (llvm::isa<WhileOp, WhileRegionOp>(op))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
these ops. This is only safe to do when compiling to XLA. }]; let constructor = "TF::CreateDropWhileShapeInvariantPass()"; } def DropWhileShapeInvariantInDeviceClusterPass : Pass<"tf-drop-while-shape-invariant-in-device-cluster", "mlir::func::FuncOp"> { let summary = "Drop `shape_invariant` attribute from While/WhileRegion ops inside device cluster."; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)