- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 63 for else_branch (0.33 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/decompose_optionals.mlir
// CHECK-LABEL: @if func.func @if() { // 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
type: DT_FLOAT type: DT_FLOAT } } } attr { key: "Tout" value { list { type: DT_FLOAT } } } attr { key: "else_branch" value { func { name: "cond_false" } } } attr { key: "then_branch" value { func { name: "cond_true" } } }
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/region-control-flow-to-functional.mlir
// CHECK-NEXT: "tf.Asin" // CHECK-NEXT: "tf.If"({{.+}}) <{else_branch = @tf.IfRegion_else, {{.+}} then_branch = @tf.IfRegion_then} // CHECK: func private @tf.IfRegion_else // CHECK-NEXT: "tf.Neg" // CHECK: func private @tf.IfRegion_then // CHECK-NEXT: "tf.Abs" func.func @testNested(%arg0: tensor<i1>, %arg1: tensor<*xf32>) -> tensor<*xf32> { // CHECK: "tf.If"({{.+}}) <{else_branch = @tf.IfRegion1_else, {{.+}} then_branch = @tf.IfRegion1_then}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 02 11:15:34 UTC 2024 - 44.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/functional-control-flow-to-cfg.mlir
func.func @testIf1Result(tensor<i1>, tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32> { ^bb0(%arg0: tensor<i1>, %arg1: tensor<*xf32>, %arg2: tensor<*xf32>): %1 = "tf.If"(%arg0, %arg1, %arg2) { then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false } : (tensor<i1>, tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32> // CHECK: [[TOBOOL:%.+]] = "tf.ToBool"(%arg0) : (tensor<i1>) -> tensor<i1>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/pin-ops-with-side-effects.mlir
func.return } // CHECK-LABEL: @tf_if_gets_control_node 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
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/transforms/decompose_reduce_dataset.cc
dataset_if->setAttr("_lower_using_switch_merge", builder.getBoolAttr(true)); // Empty else branch, if there is no more data, do nothing. auto& else_branch = dataset_if.getElseBranch(); else_branch.push_back(new Block); builder.setInsertionPointToEnd(&else_branch.front()); // Return only the state variables from the body arguments. SmallVector<Value, 4> else_returns; for (int i = 1; i < state_size + 1; i++) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
bool* has_outside_compilation) { // Instantiate "then_branch" and "else_branch". NameAttrList then_branch, else_branch; TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "then_branch", &then_branch)); TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "else_branch", &else_branch)); // Extract outside compilation for then_branch and else_branch. bool then_branch_has_outside_compilation = false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K 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
// In the newly cloned function, check that we have a _tf.If operation and capture the then and else branch. // 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]]
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
from tensorflow.python.ops import control_flow_ops # 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]]"(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 17 02:29:35 UTC 2022 - 2K bytes - Viewed (0)