- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CreateTPUResourceReadForWritePass (0.34 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
std::unique_ptr<OperationPass<ModuleOp>> CreateTPUDynamicLayoutPass(); // Creates a pass that adds `tf.ReadVariableOp` to a TPU cluster for resources // the cluster only writes to. std::unique_ptr<OperationPass<ModuleOp>> CreateTPUResourceReadForWritePass(); // Creates a pass that reorders partitiioned resource reads and replicated // inputs. std::unique_ptr<OperationPass<func::FuncOp>> CreateTPUReorderReplicateAndPartitionedInputsPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
%identity = "tf.Identity"(%arg0) : (tensor<i32>) -> tensor<i32> return %identity : tensor<i32> } ``` }]; let constructor = "TFTPU::CreateTPUResourceReadForWritePass()"; } def HoistReplicateInvariantResourceWritesPass : Pass<"tf-hoist-replicate-invariant-resource-writes", "mlir::func::FuncOp"> { let summary = "Hoists writes to replicate invariant resource variables.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)