- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 22 for then_branch (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/decompose_optionals.mlir
// CHECK-NOT: Optional %0 = builtin.unrealized_conversion_cast to tensor<5xi1> %1 = "tf.OptionalNone"() : () -> tensor<!tf_type.variant<tensor<f32>>> %2 = "tf.If"(%0, %1) <{else_branch = @false, is_stateless = false, then_branch = @true}> : (tensor<5xi1>, tensor<!tf_type.variant<tensor<f32>>>) -> (tensor<!tf_type.variant<tensor<f32>>>) return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/functional-if-ops.pbtxt
list { type: DT_FLOAT } } } attr { key: "else_branch" value { func { name: "cond_false" } } } attr { key: "then_branch" value { func { name: "cond_true" } } } experimental_debug_info { } } node { name: "StatelessIf" op: "StatelessIf" input: "tf.Less" input: "a"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 15 19:42:47 UTC 2021 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-input-shapes.pbtxt
list { type: DT_INT32 } } } attr { key: "else_branch" value { func { name: "identity_function" } } } attr { key: "then_branch" value { func { name: "identity_function" } } } experimental_debug_info { } } library { function { signature { name: "identity_function"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 11 19:14:04 UTC 2020 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/pin-ops-with-side-effects.mlir
func.func @tf_if_gets_control_node(%arg0: tensor<1xi1>)->() { "tf.If"(%arg0) {_lower_using_switch_merge = true, _read_only_resource_inputs = [], device = "", else_branch = @noop, is_stateless = false, output_shapes = [#tf_type.shape<>], then_branch = @noop} : (tensor<1xi1>) -> () func.return } // CHECK-NEXT: %[[CONTROL:.*]] = tfl.control_node controls "tf.If" // CHECK-NEXT: return // CHECK-LABEL: @tfl_if_gets_control_node
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 17 10:45:19 UTC 2022 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_tail_with_tobool_op.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 21:23:47 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/functionalize-if.mlir
// CHECK: func private @[[FUNCTIONALIZE_FUNC]] // CHECK: "tf.If" // CHECK-SAME: else_branch = @[[ELSE_FUNC:[A-Za-z0-9_]*]] // CHECK-SAME: then_branch = @[[THEN_FUNC:[A-Za-z0-9_]*]] // We expect the _tf.Add in the else func and the _tf.Mul in the then func // CHECK: func private @[[ELSE_FUNC]] // CHECK: "tf.Add" // CHECK: func private @[[THEN_FUNC]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/control_flow_upgrade_legacy_v1.py
# Tests V1 control flow is functionalized. # CHECK-NOT: tf_executor.Switch # CHECK-NOT: tf_executor.Merge # CHECK: "tf.If" # CHECK-SAME: else_branch = @"key/[[else:[a-zA-Z_0-9]+]]" # CHECK-SAME: then_branch = @"key/[[then:[a-zA-Z_0-9]+]]" # CHECK: func private @"key/[[else]]"( # CHECK-SAME: tf._original_func_name # CHECK: func private @"key/[[then]]"( # CHECK-SAME: tf._original_func_name def Test():
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 17 02:29:35 UTC 2022 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/inline.mlir
// CHECK: [[z:%.*]] = mlrt.cond [[cond]] @then @else // CHECK: return [[z]] func.func @while_cond_if(%cond: tensor<i1>, %x: tensor<i1>, %y: tensor<i1>, %z: tensor<i32>) -> (tensor<i1>) { %r = "tf.If"(%cond, %x, %y, %z) {then_branch = @then, else_branch = @else, is_stateless = true} : (tensor<i1>, tensor<i1>, tensor<i1>, tensor<i32>) -> tensor<i1> return %r : tensor<i1> } // CHECK-LABEL: func @while_body_if
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 01:01:31 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/if_op.mlir
func.func @main(%arg0: tensor<1xf32>, %arg1: 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> func.return %1 : tensor<1xf32> } func.func @cond_true(%arg0: tensor<*xf32>, %arg1: tensor<*xf32>) -> tensor<*xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 14 16:41:28 UTC 2022 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/partitioned-topological-sort.mlir
%tmp1 = "tfl.add"(%const, %const) { fused_activation_function = "NONE" } : (tensor<1xf32>,tensor<1xf32>) -> (tensor<1xf32>) %tmp2 = "tf.If"(%tmp1, %tmp1) { then_branch=@id, else_branch=@id, is_stateless = true } : (tensor<1xf32>, tensor<1xf32>) -> (tensor<1xf32>) %tmp3 = "tfl.add"(%const, %tmp2) { fused_activation_function = "NONE" } : (tensor<1xf32>, tensor<1xf32>) -> (tensor<1xf32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 19 22:33:49 UTC 2022 - 8.1K bytes - Viewed (0)