- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 68 for then_branch (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/functional-control-flow-to-regions.mlir
func.return %1 : tensor<*xf32> } // ----- func.func private @then_branch() -> () func.func private @else_branch() -> () // Test tf.If device is preserved. // CHECK-LABEL: func @testIfDevice func.func @testIfDevice(%arg0: tensor<i1>) { "tf.If"(%arg0) {then_branch = @then_branch, else_branch = @else_branch, is_stateless = false, device = "/device:CPU:0"} : (tensor<i1>) -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 06 21:59:28 UTC 2023 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/merge_tf_if_ops.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/remove_tf_if_const_args.mlir
// CHECK: [[res:%.*]] = "tf.If"([[cond]], [[x]]) // CHECK-SAME: {else_branch = @else_removed_const_args_0, is_stateless = false, then_branch = @then_removed_const_args_0} // CHECK-NEXT: return [[res]] %1 = "tf.If"(%cond, %x, %0) {else_branch = @else, then_branch = @then, is_stateless = false} : (tensor<i1>, tensor<i32>, tensor<i32>) -> tensor<i32> func.return %1 : tensor<i32> } // CHECK-LABEL: func @multiple_uses
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 10:51:48 UTC 2022 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/reorder_assert.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 10:51:48 UTC 2022 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/if_op.mlir
// CHECK: %{{.*}} = "tf.If"(%{{.*}}, %{{.*}}, %{{.*}}) <{else_branch = @cond_false, is_stateless = false, then_branch = @cond_true}> : (tensor<1xi1>, tensor<1xf32>, tensor<1xf32>) -> tensor<1xf32> %0 = "tfl.less"(%arg0, %arg1) : (tensor<1xf32>, tensor<1xf32>) -> tensor<1xi1> %1 = "tf.If"(%0, %arg0, %arg1) {else_branch = @cond_false, then_branch = @cond_true, is_stateless = false} : (tensor<1xi1>, tensor<1xf32>, tensor<1xf32>) -> tensor<1xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/optimize_tf_control_flow_side_effect.mlir
// CHECK: tf.If // CHECK-SAME: is_stateless = true %1 = "tf.If"(%cond, %arg) {else_branch = @no_side_effect_body, then_branch = @no_side_effect_body2, is_stateless = false} : (tensor<i1>, tensor<i32>) -> tensor<i32> func.return %0, %1 : tensor<i32>, tensor<i32> } // CHECK-LABEL: func @nested_set_stateless
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 01:15:55 UTC 2023 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/optimize_functional_ops.mlir
%3 = "tf.If"(%2, %0, %1) {else_branch = @sub, then_branch = @addormul, is_stateless = true} : (tensor<i1>, tensor<f32>, tensor<f32>) -> tensor<f32> func.return %3 : tensor<f32> } func.func private @addormul(%arg0: tensor<*xf32>, %arg1: tensor<*xf32>) -> tensor<*xf32> { %0 = arith.constant dense<false> : tensor<i1>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 30 10:34:48 UTC 2022 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/control_flow_duplicate_v1.py
# CHECK: "tf.If" # CHECK-SAME: else_branch = @[[else:[a-zA-Z_0-9]+]] # CHECK-SAME: then_branch = @[[then:[a-zA-Z_0-9]+]] # CHECK: func {{.*}} tf_saved_model.exported_names = ["key_2"] # CHECK: "tf.If" # CHECK-SAME: else_branch = @[[else]] # CHECK-SAME: then_branch = @[[then]] # CHECK: func private @[[else]]( # CHECK: func private @[[then]]( def Test():
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/deduplicate_if_results.mlir
// CHECK-NEXT: [[r:%.*]] = "tf.If" // CHECK-NEXT: return [[r]], [[r]] : tensor<i32>, tensor<i32> %0, %1 = "tf.If"(%cond, %x, %y) {else_branch = @else, then_branch = @then, is_stateless = true} : (tensor<i1>, tensor<i32>, tensor<i32>) -> (tensor<i32>, tensor<i32>) return %0, %1 : tensor<i32>, tensor<i32> } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jul 01 23:50:06 UTC 2023 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/functional-if-ops.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 2.5K bytes - Viewed (0)