- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 22 for then_branch (0.14 sec)
-
tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir
// CHECK: [[z:%.*]] = mlrt.cond [[cond]] @then @else([[x]], [[y]]) // CHECK: return [[z]] func.func @main(%cond: tensor<i1>, %x: tensor<i32>, %y: tensor<i32>) -> tensor<i32> { %z = "tf.If"(%cond, %x, %y) {then_branch = @then, else_branch = @else, is_stateless = true} : (tensor<i1>, tensor<i32>, tensor<i32>) -> tensor<i32> return %z: tensor<i32> } // ----- // Test lowering AsyncOpKernel // CHECK-LABEL: func @main
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_custom_aggregation_ops.mlir
%1 = "tf.Greater"(%0, %cst_0) {device = ""} : (tensor<f32>, tensor<f32>) -> tensor<i1> %2:2 = "tf.If"(%1, %arg0) <{else_branch = @cond_false_80, is_stateless = true, then_branch = @cond_true_70}> {Tcond = i1, Tin = [f32], Tout = [i1, f32], _lower_using_switch_merge = true, _read_only_resource_inputs = [], device = ""} : (tensor<i1>, tensor<1x4xf32>) -> (tensor<i1>, tensor<1x3xf32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 32.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
auto original_point = rewriter.saveInsertionPoint(); rewriter.setInsertionPointAfter(op->getParentOfType<func::FuncOp>()); SymbolTable manager(op->getParentOfType<ModuleOp>()); // Constructs `then_branch`, which is executed when `if_cond` evaluates to // true. auto then_branch_op = rewriter.create<func::FuncOp>(loc, "cond_true", func_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/quantize.mlir
%arg1: tensor<4x!quant.uniform<u8:f32, 1.0>>) -> (tensor<4x!quant.uniform<u8:f32, 1.0>>) { %0 = "tfl.dequantize"(%arg1) : (tensor<4x!quant.uniform<u8:f32, 1.0>>) -> tensor<4xf32> %1 = "tf.If"(%arg0, %0) {then_branch = @testIfThen, else_branch = @testIfElse, is_stateless = false} : (tensor<i1>, tensor<4xf32>) -> tensor<4xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:10:13 UTC 2024 - 39.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
`tf.IfRegion` and `tf.While` is transformed to `tf.WhileRegion`. For example, this functional operation ```mlir %0 = "tf.If"(%arg0, %arg1) { then_branch = @then_branch_func, else_branch = @else_branch_func, is_stateless = false } : (tensor<i1>, tensor<*xf32>) -> tensor<*xf32> ``` will be transformed into this region-based operation
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
} auto else_attr = mlir::SymbolRefAttr::get(builder.getContext(), func_names.at(else_idx)); return llvm::SmallVector<mlir::NamedAttribute, 4>{ builder.getNamedAttr("then_branch", then_attr), builder.getNamedAttr("else_branch", else_attr), // TODO(b/139667752): Analyze statelessness correctly builder.getNamedAttr("is_stateless", builder.getBoolAttr(false))}; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
LogicalResult IfOp::verifySymbolUses(SymbolTableCollection& symbol_table) { auto branch_name = [](unsigned index) -> std::string { return index == 0 ? "'then_branch'" : "'else_branch'"; }; return VerifyCaseOrIfOpBranchFunctions( symbol_table, *this, {getThenBranchAttr(), getElseBranchAttr()}, branch_name); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
analysis/analysis-api-fe10/tests-gen/org/jetbrains/kotlin/analysis/api/fe10/test/cases/generated/cases/components/expressionInfoProvider/Fe10IdeNormalAnalysisSourceModuleIsUsedAsExpressionTestGenerated.java
} @Test @TestMetadata("when_branch.kt") public void testWhen_branch() { runTest("analysis/analysis-api/testData/components/expressionInfoProvider/isUsedAsExpression/when_branch.kt"); } @Test @TestMetadata("when_branchUsed.kt") public void testWhen_branchUsed() {
Registered: Wed Jun 12 09:53:16 UTC 2024 - Last Modified: Wed May 08 19:07:43 UTC 2024 - 76.4K bytes - Viewed (0) -
analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/components/expressionInfoProvider/FirIdeDependentAnalysisSourceModuleIsUsedAsExpressionTestGenerated.java
} @Test @TestMetadata("when_branch.kt") public void testWhen_branch() { runTest("analysis/analysis-api/testData/components/expressionInfoProvider/isUsedAsExpression/when_branch.kt"); } @Test @TestMetadata("when_branchUsed.kt") public void testWhen_branchUsed() {
Registered: Wed Jun 12 09:53:16 UTC 2024 - Last Modified: Wed May 08 19:07:43 UTC 2024 - 76.4K bytes - Viewed (0)