- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 71 for then_branch (0.39 sec)
-
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/tensorflow/ir/tf_ops.td
TF_Int32Tensor:$max_num_elements ); } def TF_IfOp : TF_Op<"If", [DeclareOpInterfaceMethods<SymbolUserOpInterface>]> { let summary = "output = cond ? then_branch(input) : else_branch(input)"; let description = [{ output = cond ? then_branch(input) : else_branch(input) cond: A Tensor. If the tensor is a scalar of non-boolean type, the scalar is converted to a boolean according to the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tensor_array_ops_decomposition.mlir
// CHECK: %[[GVAR2:.*]] = "tf.MlirLocalVarOp"() : () -> tensor<!tf_type.resource<tensor<5x3xf32>>> // CHECK: "tf.If"(%[[COND]], %[[VAR]], %[[GVAR1]], %[[GVAR2]]) %1 = "tf.If"(%cond, %ta#0) { then_branch = @then_branch, else_branch = @else_branch, device = "", is_stateless = false} : (tensor<i1>, tensor<!tf_type.resource>) -> tensor<!tf_type.resource>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 49K 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/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) -
tensorflow/compiler/mlir/tensorflow/tests/resource_op_lifting.mlir
// CHECK: %[[CLUSTER:.*]]:2 = "tf_device.cluster"() %2 = "tf_device.cluster"() ({ // CHECK: %[[IF:.*]]:2 = "tf.If"(%[[ARG0]], %[[READ0]], %[[READ1]]) %3:2 = "tf.If"(%arg0, %0, %1) {then_branch = @if_then, else_branch = @if_else, is_stateless = false} : (tensor<i1>, tensor<*x!tf_type.resource<tensor<4xf32>>>, tensor<*x!tf_type.resource<tensor<4xf32>>>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 74K 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) -
tensorflow/compiler/jit/compilability_check_util.cc
if (!s.ok()) { *uncompilable_reason = s.message(); return false; } return true; } // Tests whether 'if_node' is compilable. Every operator in the then_branch and // else_branch functions must be compilable for 'if_node' to be compilable. bool RecursiveCompilabilityChecker::IsCompilableIf( const Node& if_node, FunctionLibraryRuntime* lib_runtime,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/freeze_variables.mlir
%handle = "tf.VarHandleOp"() {container="", shared_name="var1", device = "/job:worker/replica:0/task:1/device:CPU:0"} : () -> tensor<!tf_type.resource<tensor<0xf32>>> %1 = "tf.If"(%arg0, %arg1, %handle) { then_branch = @testIfThen, else_branch = @testIfElse, is_stateless = false } : (tensor<i1>, tensor<2xf32>, tensor<!tf_type.resource<tensor<0xf32>>>) -> tensor<2xf32> func.return %1 : tensor<2xf32> } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 23.6K bytes - Viewed (0)