- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 36 for statefulpartitionedcall (0.28 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util_test.cc
%0 = "tf.StatefulPartitionedCall"(%arg0) {_xla_compile_device_type = "CPU", config = "", config_proto = "", device = "/device:CPU:0", executor_type = "", f = @func} : (tensor<i32>) -> (tensor<i32>) func.return %0 : tensor<i32> } func.func @entry_func_2(%arg0: tensor<i32>) -> tensor<i32> attributes {tf_saved_model.initializer_type = ""} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/embedding_sequencing.mlir
// The order of these functions is also significant. // CHECK: {{.*StatefulPartitionedCall.* f = @_func_non_tpu.*}} // CHECK-NEXT: {{.*StatefulPartitionedCall.* f = @_func_sc_forward.*}} // CHECK-NEXT: {{.*StatefulPartitionedCall.* f = @_func_core_tpu.*}} // CHECK-NEXT: {{.*StatefulPartitionedCall.* f = @_func_sc_backward.*}} // CHECK-NEXT: return // metadata ops
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 01 21:27:49 UTC 2023 - 19.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/embedding_pipelining.mlir
// CHECK: {{.*StatefulPartitionedCall.* f = @while_cond.*}} // CHECK: {{.*StatefulPartitionedCall.* f = @non_tpu.*}} // CHECK: {{.*StatefulPartitionedCall.* f = @start_step_0.*}} // CHECK: {{.*StatefulPartitionedCall.* f = @while_cond.*}} // CHECK: {{.*StatefulPartitionedCall.* f = @non_tpu.*}} // CHECK: {{.*StatefulPartitionedCall.* f = @start_step_1.*}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 33.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_initializer_function_ops_to_main.mlir
func.func @main(%arg0: tensor<?x!tf_type.string> {tf_saved_model.index_path = ["serving_default_input_vocabs:0"]}) -> (tensor<*xi64> {tf_saved_model.index_path = ["StatefulPartitionedCall:0"]}) attributes {tf.entry_function = {inputs = "serving_default_input_vocabs:0", outputs = "StatefulPartitionedCall:0"}, tf_saved_model.exported_names = ["main"]} { %0 = tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:49:35 UTC 2023 - 29.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); Node* call; NameAttrList f_name_attr; f_name_attr.set_name(fd.signature().name()); TF_ASSERT_OK( NodeBuilder("B", "StatefulPartitionedCall", &root.graph()->flib_def()) .Input(inputs) .Attr("Tin", {DT_RESOURCE}) .Attr("Tout", {DT_RESOURCE}) .Attr("f", f_name_attr) .Finalize(root.graph(), &call));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
NodeWith(Op("Switch"), Inputs(Out(0, xla_compile), Out(1, xla_compile))); auto xla_run = NodeWith(Op("_XlaRun"), Inputs(Out(1, predicated_compilation_key))); auto tf_call = NodeWith(Op("StatefulPartitionedCall"), CtrlDeps(NodeWith(Op("Identity"), Inputs(Out(0, predicated_compilation_key))))); auto merge = NodeWith(Op("_XlaMerge"), Inputs(Out(tf_call), Out(xla_run)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/stack_ops_decomposition.mlir
// ----- // Tests PartitionedCall/StatefulPartitionedCall. // CHECK-LABEL: func @main func.func @main(%arg0: tensor<i1>) -> () { %max_size = "tf.Const"() {value = dense<10> : tensor<i32>} : () -> tensor<i32> // CHECK-NOT: tf.Stack %stack = "tf.StackV2"(%max_size) {elem_type = f32, stack_name = "s"} : (tensor<i32>) -> tensor<!tf_type.resource> // CHECK: "tf.StatefulPartitionedCall" // CHECK-SAME: f = @callee_stack_decomposed
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 25.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/sink_in_invariant_ops.mlir
// CHECK: [[handle:%.*]] = "tf.VarHandleOp"() %handle = "tf.VarHandleOp"() {container = "", shared_name = "x"} : () -> tensor<!tf_type.resource<tensor<i32>>> // CHECK: "tf.StatefulPartitionedCall"([[handle]]) %x = "tf.StatefulPartitionedCall"(%handle) {device = "/CPU:0", config = "", config_proto = "", executor_type = "", f = @some_func} : (tensor<!tf_type.resource<tensor<i32>>>) -> (tensor<i32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/hoist_invariant_ops.mlir
// CHECK-NOT: tf.VarHandleOp // CHECK: "tf._TfrtGetResource"() %handle = "tf.VarHandleOp"() {container = "", shared_name = "x"} : () -> tensor<!tf_type.resource<tensor<i32>>> // CHECK: tf.StatefulPartitionedCall %x = "tf.StatefulPartitionedCall"(%handle) {device = "/CPU:0", config = "", config_proto = "", executor_type = "", f = @some_func} : (tensor<!tf_type.resource<tensor<i32>>>) -> (tensor<i32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 01 23:54:14 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/components/tf_to_stablehlo.mlir
// ----- // This test makes sure functions with tf._noinline=true is not inlined. module { func.func @stateful_partitioned_call(%arg0: tensor<1x2x2x3xf32>) -> (tensor<1x2x2x3xf32>) { %0 = "tf.StatefulPartitionedCall"(%arg0) <{ config = "", config_proto = "", executor_type = "", f = @some_func }> { _collective_manager_ids = [], device = ""
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 20:05:12 UTC 2024 - 13.6K bytes - Viewed (0)