- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TPUClusterFormationPass (0.4 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
#define GEN_PASS_DEF_TPUCLUSTERFORMATIONPASS #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc" class TPUClusterFormationPass : public impl::TPUClusterFormationPassBase<TPUClusterFormationPass> { public: explicit TPUClusterFormationPass(bool strict_clusters) : strict_clusters_(strict_clusters) {} void getDependentDialects(DialectRegistry& registry) const override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.td
compiled, the result is device agnostic and in TF functional dialect & that the device attribute exists. }]; let constructor = "tensorflow::tf2xla::internal::CreateVerifyClusteringPass()"; } def TPUClusterFormationPass : Pass<"tf-tpu-cluster-formation", "ModuleOp"> { let summary = "Forms clusters from operations assigned to the same TPU computation"; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 19.8K bytes - Viewed (0)