- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for CreateTPUResourceReadForWritePass (0.31 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_read_for_write.cc
func.getResultTypes())); cluster_func.erase(); } } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateTPUResourceReadForWritePass() { return std::make_unique<TPUResourceReadForWritePass>(); } } // namespace TFTPU
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 16:54:40 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
pm.addNestedPass<FuncOp>( tensorflow::tf2xla::internal::CreateXlaBroadcastPass()); pm.addPass(mlir::TFDevice::CreateClusterOutliningPass()); pm.addPass(mlir::TFTPU::CreateTPUResourceReadForWritePass()); pm.addPass(mlir::TFDevice::CreateMarkInputOutputAliasesPass()); pm.addPass( tensorflow::tf2xla::internal::CreateTPUShardingIdentificationPass()); pm.addNestedPass<FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 16:09:14 UTC 2024 - 11.2K bytes - Viewed (1) -
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)