- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 43 for IfRegion (0.3 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// However, its probably better to fold the IfRegion instead of having the // dead branch stay. // Inline the region in place of the IfRegion op, and forward the yield // inputs to the IfRegion op results. This is possible only if the yield // types match the result types. auto yield = cast<YieldOp>(region.front().getTerminator()); auto updated_results = llvm::to_vector<4>(yield.getOperands());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir
func.return %graph : tensor<i1> // expected-remark@above {{ID: 4}} // expected-remark@above {{Sinks: {}}} } // ----- // Tests that the pass does not add control dependencies for a stateless // IfRegion op. // CHECK-LABEL: func @stateless_ifregion_op func.func @stateless_ifregion_op( // expected-remark@above {{ID: 18}} %arg0: tensor<*x!tf_type.resource<tensor<32xf32>>>, %arg1: tensor<i1>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 129.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/freeze_variables.mlir
// CHECK-NOT: "tf.VarHandleOp" %handle = "tf.VarHandleOp"() {container="", shared_name="var1", device = "/job:worker/replica:0/task:1/device:CPU:0"} : () -> tensor<!tf_type.resource<tensor<0xf32>>> %0 = "tf.IfRegion"(%arg0) ({ %1 = "tf.ReadVariableOp"(%handle) : (tensor<!tf_type.resource<tensor<0xf32>>>) -> (tensor<0xf32>) "tf.Yield"(%1) : (tensor<0xf32>) -> () }, {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
%true = "tf.Const"() {value = dense<true> : tensor<i1>} : () -> tensor<i1> // CHECK: [[Val0:%.*]] = "tf.Mul"(%arg0, %arg1) // CHECK-SAME: device = "/job:localhost/replica:0/task:0/device:GPU:0" %0 = "tf.IfRegion"(%true) ({ %true_value = "tf.Mul"(%arg0, %arg1) {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<f32>, tensor<f32>) -> tensor<f32> "tf.Yield"(%true_value) : (tensor<f32>) -> () }, {
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/tensorflow/tests/constant-fold.mlir
// preserved. // // CHECK-LABEL: func @yieldOp func.func @yieldOp(%arg0: tensor<f32>, %arg1: tensor<f32>, %arg2: tensor<i1>) -> (tensor<f32>) { // CHECK-COUNT-2: tf.Yield %0 = "tf.IfRegion"(%arg2) ({ "tf.Yield"(%arg0) : (tensor<f32>) -> () }, { "tf.Yield"(%arg1) : (tensor<f32>) -> () }) { is_stateless = true}: (tensor<i1>) -> tensor<f32> func.return %0 : tensor<f32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 31 23:22:24 UTC 2024 - 36.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir
// CHECK: %[[BUFFER:.*]] = "tf.BroadcastTo"(%[[ZERO_F32]], %[[MAX_SIZE]]) // CHECK: %[[BUFFER_SIZE:.*]] = "tf.Const"() <{value = dense<0> : tensor<1xi32>}> // CHECK-NOT: tf.EmptyTensorList %if_op = "tf.IfRegion"(%arg0) ({ %elem = "tf._SomeOp"() : () -> tensor<f32> // CHECK: %[[UPDATE:.*]] = "tf.XlaDynamicUpdateSlice" // CHECK: %[[ONE:.*]] = "tf.Const"() <{value = dense<1> : tensor<1xi32>}> : () -> tensor<1xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
std::string retvals_communication_key = llvm::formatv("host_compute_channel_{0}_retvals", (communication_key_index)) .str(); // Use a unique name when sending just the IfRegion predicate. This is // for readable and to match the key in the TF2XLA bridge. if (clustered_ops.size() == 1 && llvm::isa<mlir::TF::IfRegionOp>(op) && external_operands.size() == 1) { args_communication_key =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/fuse-tftext.mlir
%18 = "tf.Equal"(%17, %4) {device = "", incompatible_shape_error = true} : (tensor<i64>, tensor<i64>) -> tensor<i1> %19 = "tf.All"(%18, %5) {device = "", keep_dims = false} : (tensor<i1>, tensor<0xi32>) -> tensor<i1> %20 = "tf.IfRegion"(%19) ({ %72 = "func.call"(%19, %17, %4) {callee = @RaggedFromNestedRowSplits_RaggedFromRowSplits_RowPartitionFromRowSplits_assert_equal_1_Assert_AssertGuard_true_27770} : (tensor<i1>, tensor<i64>, tensor<i64>) -> tensor<i1>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 460.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
bool InferShapeForRestore(Operation* op); // Infers the shape IfOp outputs based on the shapes of the then and else // function result types. bool InferShapeForIf(IfOp op); // Infers the shape IfRegion outputs based on the shapes of the then and else // yields. bool InferShapeForIfRegion(IfRegionOp op); // Infers the shape CaseOp outputs based on the shapes of branch function // result types.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
types of inputs must match the signature of the operation that contains the region. }]; let arguments = (ins Variadic<AnyType>); } def TF_IfRegionOp : TF_Op<"IfRegion", [SingleBlockImplicitTerminator<"YieldOp">, NoRegionArguments, DeclareOpInterfaceMethods<RegionBranchOpInterface, [ "areTypesCompatible", "getEntrySuccessorOperands",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0)