- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 20 for StatefulPartitionedCall (0.72 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tf-variables.mlir
(tensor<1x10xf32> {tf_saved_model.index_path = ["output_0"]}, tensor<1x10xi64> {tf_saved_model.index_path = ["output_1"]}) attributes {tf.entry_function = {control_outputs = "", inputs = "serving_default_x:0", outputs = "StatefulPartitionedCall:0"}, tf_saved_model.exported_names = ["serving_default"]} { %handle_0 = "tf.VarHandleOp"() {container="c", shared_name="a"} : () -> tensor<!tf_type.resource<tensor<1x10xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/call_to_exported.py
# CHECK-SAME: tensor<f32> {tf_saved_model.index_path = [1]}) # CHECK-SAME: attributes{{.*}}tf_saved_model.exported_names = ["callee"] # CHECK: "tf.StatefulPartitionedCall"{{.*}}f = @[[CALLEE_INTERNAL:[a-zA-Z_0-9]+]] # # CHECK: func {{@[a-zA-Z_0-9]+}}( # CHECK-SAME: %arg0: tensor<f32> {tf._user_specified_name = "x", tf_saved_model.index_path = [0]},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 28 19:09:38 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_validate_inputs.mlir
func.func @nested_entry_functions() attributes {tf.entry_function = {}} { tf_executor.graph { %control = tf_executor.island wraps "tf.StatefulPartitionedCall"() {config = "", config_proto = "", device = "/device:CPU:0", executor_type = "", f = @func} : () -> () tf_executor.fetch } func.return } func.func @func() attributes {tf.entry_function = {}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 19:29:14 UTC 2024 - 818 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
::llvm::cl::desc("Skip tf.ResizeBilinear and tf.ResizeNearestNeighbor")}; Option<bool> skip_partitioned_calls_{ *this, "skip-partitioned-calls", ::llvm::cl::desc( "Skip tf.StatefulPartitionedCall and tf.PartitionedCall")}; }; void TFToMhloPass::runOnOperation() { auto func = getOperation(); MLIRContext *context = func->getContext(); RewritePatternSet patterns(context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_freeze_global_tensors.mlir
func.func @f(%arg0: tensor<!tf_type.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) attributes {tf_saved_model.exported_names = ["f"]} { "tf.StatefulPartitionedCall"(%arg0) {config = "", config_proto = "", executor_type = "", f = @f_callee} : (tensor<!tf_type.resource<tensor<f32>>>) -> () func.return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 03:07:35 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/sink_variable_as_named_array.mlir
// // CHECK-LABEL: func.func @serving_default // CHECK-NOT: IfrtLoadVariable // CHECK: "tf.VarHandleOp" // CHECK-NEXT: "tf.AssignVariableOp" // CHECK-NEXT: "tf.ReadVariableOp" // CHECK-NEXT: "tf.StatefulPartitionedCall" // CHECK-NEXT: return // module { func.func @serving_default() -> tensor<*xi32> { %cst = "tf.Const"() <{value = dense<"some_test.txt"> : tensor<!tf_type.string>}> : () -> tensor<!tf_type.string>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 15:33:17 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.h
// When add_edges_to_output_of_downstream_nodes is true, the output edges of // the xla_launch_node's immediate downstream nodes would be attached to the // generated xla node. For example, if the original graph is // StatefulPartitionedCall{_xla_compile_id=1} -> XlaClusterOutput -> NodeA // The output graph of this function would look like the following when // add_edges_to_output_of_downstream_nodes is true: // XlaLaunch -> NodeA
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/end-to-end-tpu-reshard-variables.mlir
tf_executor.graph { %control = tf_executor.island { "tf.StatefulPartitionedCall"(%arg0) <{config = "", config_proto = "", executor_type = "", f = @partitioned}> : (tensor<*x!tf_type.resource>) -> () tf_executor.yield } tf_executor.fetch %control : !tf_executor.control }
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/tests/tf_saved_model_freeze_assets.mlir
// CHECK: func @f(%arg0 func.func @f(%arg0: tensor<!tf_type.string> {tf_saved_model.bound_input = @v}) attributes {tf_saved_model.exported_names = ["f"]} { "tf.StatefulPartitionedCall"(%arg0) {config = "", config_proto = "", executor_type = "", f = @f_callee} : (tensor<!tf_type.string>) -> () func.return } func.func private @f_callee(%arg0: tensor<!tf_type.string>) { func.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
bool IsNonReplicatedGraph(const Graph& graph, const FunctionLibraryDefinition* function_library) { auto predicate = [](const Graph& graph) { const std::string kStatefulPartitionedCallOp = "StatefulPartitionedCall"; for (const Node* node : graph.nodes()) { auto node_op = node->type_string(); if (node_op == kStatefulPartitionedCallOp) { // Functions called by StatefulfulPartitionedCall ops with
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0)