- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 71 for then_branch (0.22 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/end2end/if_op.pbtxt
# CHECK-DAG: constant dense<[5.000000e+00, 6.000000e+00, 7.000000e+00, 8.000000e+00]> # CHECK: "tf.If"{{.+}}else_branch = @cond_false_10{{.+}}is_stateless = true{{.+}}then_branch = @cond_true_10 # CHECK: "tf.If"{{.+}}else_branch = @cond_false0{{.+}}is_stateless = false{{.+}}then_branch = @cond_true0 # CHECK: func private @cond_false_10 # CHECK-NEXT: tfl.div # CHECK: func private @cond_true_10 # CHECK-NEXT: tfl.sub
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 16 15:29:56 UTC 2021 - 6.7K 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/graphdef2mlir/graph-function-defs.pbtxt
# Verify that we properly import call site function attributes. # CHECK: tf.If # CHECK-SAME: then_branch = @ # CHECK-SAME: then_branch.how_many = 32 # CHECK-SAME: then_branch.ping = "ack" node { name: "Placeholder" op: "Placeholder" attr { key: "dtype" value { type: DT_INT32 } } attr { key: "shape"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 24 00:20:25 UTC 2020 - 7.1K 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)