Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for cluster_func (0.12 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir

    // Tests `tf_device.cluster_func` with missing `step_marker_location` attribute.
    
    module attributes {tf.versions = {producer = 888 : i32}, tf.devices = ["/job:worker/replica:0/task:0/device:CPU:0", "/job:worker/replica:0/task:0/device:TPU_SYSTEM:0", "/job:worker/replica:0/task:0/device:TPU:0"]} {
      func.func @bad_num_cores_per_replica() {
        // expected-error@+1 {{requires attribute 'step_marker_location'}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 172.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

        This pass outlines the body of a `tf_device.cluster` into a function and
        replaces the `tf_device.cluster` op with an equivalent `tf_device.cluster_func`
        op. Implicit operands will be captured and materialized as explicit arguments to
        the newly created functions and associated `tf_device.cluster_func` ops.
    
        For example, the following:
    
        ```mlir
        func @computation(%arg0: tensor<i32>) -> tensor<i32> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
Back to top