- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for PartitionResourceReadsWrites (0.48 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
// Here the unpartitioned resource read and write is expanded to individual // resource reads and writes per associated partitioned resource handle. LogicalResult PartitionResourceReadsWrites( tf_device::ClusterFuncOp cluster_func) { bool use_spmd = false; if (auto use_spmd_attr = cluster_func->getAttrOfType<BoolAttr>(kUseSpmdAttr)) use_spmd = use_spmd_attr.getValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 11.8K bytes - Viewed (0)