- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for cluster_func (0.21 sec)
-
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) -
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)