Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for BuildLaunchForCluster (0.2 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc

        }
      }
    }
    
    // Build a `tf_device.launch` op with a region that contains all the operations
    // in given cluster. Then all ops in cluster are replaced by `tf_device.launch`.
    void BuildLaunchForCluster(const TF::Cluster& c, OpBuilder* builder) {
      // Set insertion point to right after all operations in cluster.
      builder->setInsertionPoint(c.ops.back()->getNextNode());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 05 13:30:21 UTC 2023
    - 6.8K bytes
    - Viewed (0)
Back to top