Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for send_key (0.24 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/prepare_tpu_computation_for_tf_export.mlir

      "tf.XlaSendToHost"(%0) {key = "send_key"} : (tensor<i32>) -> ()
    
      // CHECK: _xla_original_oc_node_name = [[NODE_NAME3:.*]], _xla_token_input_nodes = {{\[}}[[NODE_NAME2]]{{\]}}
      %1 = "tf._XlaHostComputeMlir"(%0) {recv_key = "host_compute_channel_recv1", send_key = "host_compute_channel_send1", host_mlir_module = ""} : (tensor<i32>) -> (tensor<f32>)
      func.return
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 18:46:36 UTC 2024
    - 9.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/optional_symbol_ref.mlir

          %control = tf_executor.island wraps "tf.XlaHostCompute"() {_xla_original_oc_node_name = "Add", _xla_token_input_nodes = ["_xla_token_arg_node"], ancestors = [], cost_estimate_ns = 1024 : i64, key = "host_compute_channel_1_retvals", send_key = "", recv_key = "", shapes = [], tpu_core = 0 : i64} : () -> ()
          tf_executor.fetch
        }
        func.return
      }
    }
    
    // CHECK: op: "XlaHostCompute"
    
    // CHECK:       attr {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 28 12:06:33 UTC 2022
    - 967 bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/tf_xla_passes.td

        flow ops when necessary.
    
        For example, given the program
    
        ```mlir
          func @send_to_host(%arg0: tensor<i32>) {
            "tf.XlaSendToHost"(%arg0) {key = "send_key"} : (tensor<i32>) -> ()
            return
          }
        ```
    
        This might be legalized like this:
    
        ```mlir
          func @send_to_host(%arg0: tensor<i32>) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 03 05:56:39 UTC 2023
    - 1.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/api/v1/testdata/prepare_to_library.mlir

        return
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 31 23:44:50 UTC 2024
    - 2.3K bytes
    - Viewed (0)
Back to top