- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ReplicateInvariantOpHoistingPass (0.29 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc
#define GEN_PASS_DEF_REPLICATEINVARIANTOPHOISTINGPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" struct ReplicateInvariantOpHoistingPass : public impl::ReplicateInvariantOpHoistingPassBase< ReplicateInvariantOpHoistingPass> { void runOnOperation() override; }; // Check if op directly uses a key in `virtual_devices`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
}]; // Required for mhlo bounded shape extension. let dependentDialects = ["mhlo::MhloDialect"]; let constructor = "TFTPU::CreateTPUAnnotateDynamicShapeInputsPass()"; } def ReplicateInvariantOpHoistingPass : Pass<"tf-replicate-invariant-op-hoisting", "mlir::func::FuncOp"> { let summary = "Hoists replicate invariant operations out of replicate"; 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)