- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 19 for opA (0.02 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/replicate_to_island.mlir
} tf_executor.fetch } func.return } // CHECK: "tf.opA" // device = "/TPU:0" // CHECK: _parallel_execution_ids = "r0:0" // CHECK: "tf.opA" // device = "/TPU:0" // CHECK: _parallel_execution_ids = "r0:1" // CHECK: "tf.opA" // device = "/TPU:1" // CHECK: _parallel_execution_ids = "r1:0" // CHECK: "tf.opA" // device = "/TPU:1"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/embedding_program_key.mlir
func.func @reorder_single_op_program_key() { // CHECK: %[[COMPILE_LAUNCH:[0-9]*]]:2 = "tf_device.launch" // CHECK: TPUCompileMlir // CHECK: "tf.OpA"(%[[COMPILE_LAUNCH]]#1 "tf_device.launch"() ({ %cst_0 = "tf.Const"() {value = dense<""> : tensor<1x!tf_type.string>} : () -> tensor<1x!tf_type.string> "tf.OpA"(%cst_0) { mini_batch_splits = ""} : (tensor<1x!tf_type.string>) -> () tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 14:28:22 UTC 2023 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/replicate_invariant_op_hoisting.mlir
%2 = "tf.opA"(%1, %ri) : (tensor<?xi32>, tensor<*xf32>) -> tensor<*xi32> tf_device.return %1, %2 : tensor<?xi32>, tensor<*xi32> } func.return } // CHECK: %[[SHAPE:[0-9]*]] = "tf.Shape"(%[[ARG_0]]) // CHECK: tf_device.replicate([%[[ARG_0]], %[[ARG_1]]] as %[[RI:[a-z0-9]*]]: tensor<*xf32>) // CHECK: %[[OP_A:[0-9]*]] = "tf.opA"(%[[SHAPE]], %[[RI]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_validate_inputs.mlir
%out, %c1 = tf_executor.island wraps "tf.opA"(%ri) {_tpu_replicate="cluster"}: (tensor<i32>) -> tensor<i32> // expected-warning @+1 {{TF2XLA TPU bridge input check: cluster op = tf.opA with cluster = cluster has successor as non cluster op tf.opB}} %out2, %c2 = tf_executor.island wraps "tf.opB"(%out) : (tensor<i32>) -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 06:51:01 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/hoist_loop_invariant.mlir
// CHECK: [[RES_1:%.*]] = "tf.Add"([[CST_1]], [[CST_0]]) // CHECK: [[RES_2:%.*]] = "tf.Mul"([[RES_1]], [[CST_1]]) // CHECK: tf.WhileRegion // CHECK: ^bb0 // CHECK: tf.OpA // CHECK: ^bb0([[ARG_2:%[a-zA-Z0-9_]+]] // CHECK-SAME: [[ARG_3:%[a-zA-Z0-9_]+]]: tensor<i32>) // CHECK-NEXT: [[RES_3:%.*]] = "tf.AddV2"([[ARG_2]], [[RES_1]]) // CHECK-NEXT: [[RES_4:%.*]] = "tf.Div"([[ARG_3]], [[RES_2]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 22 17:12:02 UTC 2023 - 14.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_passes.td
For example, the following: ```mlir func @single_op_launch() { %a = "tf.opA"() {device = "CPU:0"} : () -> tensor<i1> return %a } ``` will be transformed into: ```mlir func @single_op_launch() { %1 = tf_device.launch() ( { %a = "tf.opA"() : () -> tensor<i1> tf_device.return %a }) {device = "CPU:0"} : () -> tensor<i1>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.h
// // Two ops with the same groups and different branches are considered // parallel so are not made dependent. For example if `OpA` has parallel_ids // `{group0:branch0, group1:branch0}` // and `OpB` has parallel_ids // `{group0:branch1, graph1:branch0}` // then `OpA` and `OpB` are executed in parallel because `group0` is common // with a different branch. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_resource_partitioning.mlir
%1 = "tf.ReadVariableOp"(%0) : (tensor<!tf_type.resource<tensor<i32>>>) -> tensor<i32> // CHECK: [[COMPUTATION:%.+]] = "tf_device.parallel_execute"() // CHECK: "tf.OpA"([[READ0]]) %2 = "tf_device.parallel_execute"() ({ "tf_device.launch"() ({ "tf.OpA"(%1) : (tensor<i32>) -> () tf_device.return }) {device = "TPU_REPLICATED_HOST_0"} : () -> () tf_device.return }, {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_device_ops_invalid.mlir
"tf_device.parallel_execute"() ({ // expected-error@-1 {{'tf_device.parallel_execute' op number of output types (3) must match the total number of outputs from all regions (2).}} %0 = "tf.opA"() : () -> (tensor<*xi1>) %1 = "tf.opB"() : () -> (tensor<*xi32>) tf_device.return %0, %1 : tensor<*xi1>, tensor<*xi32> }, { %2 = "tf.opC"() : () -> (tensor<*xi1>) tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 9.8K bytes - Viewed (0) -
cmd/config-current.go
case config.PolicyOPASubSys: // In case legacy OPA config is being set, we treat it as if the // AuthZPlugin is being set. subSys = config.PolicyPluginSubSys fallthrough case config.PolicyPluginSubSys: if ppargs, err := polplugin.LookupConfig(s, GetDefaultConnSettings(), xhttp.DrainBody); err != nil { return err } else if ppargs.URL == nil { // Check if legacy opa is configured.
Registered: Sun Jun 16 00:44:34 UTC 2024 - Last Modified: Thu May 30 08:14:58 UTC 2024 - 30.8K bytes - Viewed (0)