- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 25 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/tensorflow/tests/region-control-flow-to-functional.mlir
// 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} %0 = "tf.IfRegion"(%arg0) ({
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/tensorflow/transforms/decompose_reduce_dataset.cc
/*operands=*/else_returns); // Then branch gets the data and calls the reduce_function. auto& then_branch = dataset_if.getThenBranch(); then_branch.push_back(new Block); builder.setInsertionPointToEnd(&then_branch.front()); // Add iterator operational data access inside if. auto get_value = builder.create<TF::OptionalGetValueOp>(loc, dataset_types,
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/mlir/tfrt/tests/sink_in_invariant_ops.mlir
// CHECK: [[cond:%.*]] = "tf.Const"() %cond = "tf.Const"() {device = "/CPU:0", value = dense<true> : tensor<i1>} : () -> tensor<i1> // CHECK: "tf.If"([[cond]], [[handle]]) %x = "tf.If"(%cond, %handle) {then_branch = @some_other_func, else_branch = @some_other_func, is_stateless = false} : (tensor<i1>, tensor<!tf_type.resource<tensor<i32>>>) -> tensor<i32> %r = "tf.AddV2"(%arg, %x) {device = "/CPU:0"} : (tensor<i32>, tensor<i32>) -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
} else if (auto if_op = llvm::dyn_cast<TF::IfOp>(use.getOwner())) { auto then_branch = if_op.then_function(); auto else_branch = if_op.else_function(); assert(then_branch && else_branch); auto type_from_then = GetElementTypeFromAccess( then_branch.getArgument(use.getOperandNumber() - 1), module, infer_from_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
llvm::StringMap<PartitionedCallDecompositionInfo>* decomposed_partitioned_call_callees) { // Rewrite the branches. Region& then_branch = if_op.getThenBranch(); Region& else_branch = if_op.getElseBranch(); if (failed(DecomposeTensorListOpsInternal( &then_branch.front(), module, buffer_to_size, decomposed_partitioned_call_callees))) return failure(); if (failed(DecomposeTensorListOpsInternal(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/control_flow.mlir
// CHECK: [[cond_res:%.*]]:2 = tfrt.cond [[cond]] // CHECK-SAME: @cond_true @cond_false(%arg0, %arg2) : (!tfrt.chain, !tfrt_fallback.tf_tensor) %2 = "tf.If"(%arg0, %arg1) {else_branch = @cond_false, then_branch = @cond_true, is_stateless = true} : (tensor<i1>, tensor<i32>) -> tensor<i32> // CHECK: [[out_ch:%.*]] = tfrt.merge.chains [[cond_res]]#0, %arg0 : !tfrt.chain, !tfrt.chain
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 00:40:32 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
// and else_branch, and it has `recv_if_pred_node` as cond input. Node *if_oc_node = node_name_index["oc_if_if"]; EXPECT_NE(if_oc_node, nullptr); Node *if_oc_node_cond_input; TF_CHECK_OK(if_oc_node->input_node(0, &if_oc_node_cond_input)); EXPECT_EQ(if_oc_node_cond_input, recv_if_pred_node); // Check that then_branch outside compilation has node "identity_true_fn".
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
#include "tensorflow/core/public/version.h" namespace tensorflow { TEST(RearrangeFunctionArgumentForFunctionTest, Basic) { FunctionDefLibrary fdl; { // Function for StatefulPartitionedCall's "f", If's // "then_branch"/"else_branch". // "arg0" (T=DT_RESOURCE), "arg1" (T=DT_BOOL) // "ret0" = "arg1" // "ret1" = "arg0" tensorflow::Scope s = tensorflow::Scope::NewRootScope();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0)