- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for while_node (0.14 sec)
-
tensorflow/compiler/jit/compilability_check_util.cc
is_compilable &= ExtractNodeDefAndCheckCompilability( while_node, "cond", "while_cond", encapsulating_function, lib_runtime, stack_trace, uncompilable_nodes); if (!uncompilable_nodes && !is_compilable) return is_compilable; is_compilable &= ExtractNodeDefAndCheckCompilability( while_node, "body", "while_body", encapsulating_function, lib_runtime, stack_trace, uncompilable_nodes);
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/jit/extract_outside_compilation_pass.cc
TF_RETURN_IF_ERROR(GraphToFunctionDef(*g, new_name, &replace_fdef)); TF_RETURN_IF_ERROR(fld->AddFunctionDef(replace_fdef)); loop_cond_func->set_name(new_name); while_node->ClearAttr("cond"); while_node->AddAttr("cond", *loop_cond_func); } return absl::OkStatus(); } // Rewrites while loop cond function for host. Status RewriteHostWhileLoopCond(
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/transforms/convert_control_to_data_outputs.cc
AppendFunctionArguments(while_body, num_chains, chaining_data_type); AppendFunctionResults(while_body, num_chains, chaining_data_type); // Insert identity ops with control dep ChainResourceOps(while_body, chain_resource_to_ops_map, resource_equivalence_classes, ops_connected_to_fetch, num_old_outputs); // Modify all the while ops referencing the body function and the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf-while.mlir
// CHECK: yield // CHECK: ^bb0([[ARGS]]): // CHECK: call @while_body // CHECK: yield // CHECK: while_body // CHECK: while_cond // INLINE: tfl.while // INLINE: ^bb0([[ARGS:.*]]): // INLINE: tf.Less // INLINE: yield // INLINE: ^bb0([[ARGS]]): // INLINE: %cst_2 = arith.constant // INLINE: yield // INLINE-NOT: while_body // INLINE-NOT: while_cond // CANON-LABEL: func @while_main
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/end-to-end-tpu-reshard-variables.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 21:23:47 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_passes.td
```mlir func @cluster_with_loop() -> () { %0 = "tf.VarHandleOp"() ... "tf_device.cluster"() ( { %1 = "tf.While"(%0) {body = @while_body, cond = @while_cond} tf_device.return }) return } func @while_body(%arg0: tensor<*x!tf_type.resource<tensor<f32>>>) { %constant = "tf.Const"() ... "tf.AssignVariableOp"(%arg0, %constant) return %arg0 }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/while_to_map_fn.mlir
%2 = "tf.LogicalAnd"(%0, %1) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<i1>, tensor<i1>) -> tensor<i1> return %2 : tensor<i1> } // CHECK-LABEL: map/while_body func.func private @"map/while_body"(%arg0: tensor<i32>, %arg1: tensor<i32>, %arg2: tensor<!tf_type.variant<tensor<*xf32>>>, %arg3: tensor<?xf32>) -> (tensor<i32>, tensor<i32>, tensor<!tf_type.variant<tensor<*xf32>>>, tensor<?xf32>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:40:22 UTC 2024 - 68.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/tfl_while_outline.mlir
} // CHECK-LABEL: func @rnn( // CHECK: tfl.while // CHECK: tfl.yield // CHECK-SAME: (tensor<i1>) -> () // CHECK: [[VAL_30:%.*]]:7 = // CHECK: call @tfl.while_body // CHECK: tfl.yield // CHECK-SAME: (tensor<i32>, tensor<i32>, tensor<*xf32>, tensor<4x2xf32>, tensor<4x2xf32>, tensor<*xf32>, tensor<4x4x3xf32>) -> () // CHECK-LABEL: func private @tfl.while_cond(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 13.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc
erase_indices.set(i); } else if (results[i].getDefiningOp()) { FindConsumedOp(results[i].getDefiningOp(), consumed_ops); } } // Remove elements and operations in while_body that are not indispensable. yield_op.eraseOperands(erase_indices); // Remove ops from bottom to top. for (Operation &op : llvm::make_early_inc_range(reverse(body.getOperations())))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
variable ops, thus allowing them to run in parallel across iterations. Before: ```mlir !tf_res = type tensor<!tf_type.resource<tensor<f32>>> func @while_body(%arg0: !tf_res, %arg1: !tf_res, %arg2: tensor<f32>, %arg3: tensor<f32>) -> (!tf_res, !tf_res, tensor<f32>, tensor<f32>) { %graph:4 = tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)