- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for DropWhileShapeInvariantAttr (0.39 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/drop_while_shape_invariant.cc
void runOnOperation() override; }; void DropWhileShapeInvariantAttr(Operation* op) { if (llvm::isa<WhileOp, WhileRegionOp>(op)) op->removeAttr(kShapeInvariantAttr); } void DropWhileShapeInvariantPass::runOnOperation() { getOperation().walk([](Operation* op) { DropWhileShapeInvariantAttr(op); }); } void DropWhileShapeInvariantInDeviceClusterPass::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 2.4K bytes - Viewed (0)