- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 73 for statefulpartitionedcall (0.23 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/lite/stablehlo/tests/legalize-skip-partitioned-calls.mlir
%0 = "tf.StatefulPartitionedCall"(%arg0) <{ config = "", config_proto = "", executor_type = "", f = @some_func }> { _collective_manager_ids = [], device = "" } : (tensor<1x2x2x3xf32>) -> tensor<1x2x2x3xf32> // CHECK-SKIP: tf.StatefulPartitionedCall // CHECK-NOSKIP: call @some_func // CHECK-NOSKIP-NOT: tf.StatefulPartitionedCall %1 = "tf.PartitionedCall"(%0) <{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 20:05:12 UTC 2024 - 1.5K 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/tensorflow/tests/xla_cluster_formation.mlir
// CHECK: tf_device.cluster // CHECK-NEXT: tf.StatefulPartitionedCall // CHECK-NEXT: tf_device.return // CHECK: _cluster_outlined_function_name = "stateful_pcall_func" // CHECK-SAME: allow_soft_placement = true // CHECK: tf.Const // CHECK: tf.Add func.func @xla_must_compile_true(%arg0: tensor<i32>) -> tensor<i32> attributes {tf.entry_function = {}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/inlining.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-as-function.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 24 00:18:34 UTC 2023 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/signature_def.mlir
// CHECK-NEXT: buffer: 6, // CHECK-NEXT: name: "StatefulPartitionedCall:0", // CHECK-NEXT: quantization: { // CHECK-EMPTY: // CHECK-NEXT: }, // CHECK-NEXT: shape_signature: [ -1, 5 ], // CHECK-NEXT: has_rank: true // CHECK-NEXT: }, { // CHECK-NEXT: shape: [ 1, 5 ], // CHECK-NEXT: buffer: 7, // CHECK-NEXT: name: "StatefulPartitionedCall:1", // CHECK-NEXT: quantization: { // CHECK-EMPTY:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 06 18:55:51 UTC 2023 - 4.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/disallow_stateful_partitioned_call.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 24 20:05:09 UTC 2023 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize_compile_and_replicate_attributes.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 3.1K bytes - Viewed (0)