Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for IdentifyXlaShardingForTPUComputation (0.28 sec)

  1. tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc

      return absl::OkStatus();
    }
    
    // Extracts input/output sharding configuration of `cluster_func` by parsing
    // XlaSharding ops inside the `cluster_func`.
    LogicalResult IdentifyXlaShardingForTPUComputation(
        Builder* builder, mlir::tf_device::ClusterFuncOp cluster_func) {
      // Look up function definition from module.
      mlir::func::FuncOp func =
          cluster_func->getParentOfType<ModuleOp>()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 02:01:13 UTC 2024
    - 28.9K bytes
    - Viewed (0)
Back to top