- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 32 for StatefulPartitionedCall (0.29 sec)
-
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/signature.mlir
// CHECK-SAME: {tf_saved_model.index_path = ["end_logits"]} // CHECK-SAME: tf.entry_function = {inputs = "serving_default_input2:0,serving_default_input1:0", outputs = "StatefulPartitionedCall:1,StatefulPartitionedCall:0"} // CHECK-SAME: tf_saved_model.exported_names = ["serving_default"] module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, producer = 554 : i32}, tf_saved_model.semantics} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 07:35:24 UTC 2022 - 1.9K 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/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/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/mlir2graphdef/graph-as-function.mlir
// CHECK: node { // CHECK-NEXT: name: "identity" // CHECK-NEXT: op: "Identity" // CHECK-NEXT: input: "const" // CHECK: node { // CHECK-NEXT: name: "statefulpartitionedcall" // CHECK-NEXT: op: "StatefulPartitionedCall" // CHECK-NEXT: input: "const" // CHECK-NEXT: input: "args_1" // CHECK: key: "f" // CHECK-NEXT: value { // CHECK-NEXT: func {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_inline_device_ops.mlir
// CHECK-LABEL: func @simple_stateful_partitioned_call // CHECK-NOT: "tf_device.cluster" // CHECK: "tf.StatefulPartitionedCall" // CHECK-NEXT: "tf.Const" func.func @simple_stateful_partitioned_call(%arg0: tensor<i32>) -> tensor<i32> { %0 = "tf_device.cluster"() ({ %2 = "tf.StatefulPartitionedCall"(%arg0) {config = "", config_proto = "", executor_type = "", f = @stateful_pcall_func} : (tensor<i32>) -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 14:24:59 UTC 2022 - 3.5K 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_with_no_inputs.mlir
// CHECK-NEXT: version: 3, // CHECK-NEXT: operator_codes: [ ], // CHECK-NEXT: subgraphs: [ { // CHECK-NEXT: tensors: [ { // CHECK-NEXT: shape: [ 5 ], // CHECK-NEXT: buffer: 1, // CHECK-NEXT: name: "StatefulPartitionedCall:1", // CHECK-NEXT: quantization: { // CHECK-EMPTY: // CHECK-NEXT: }, // CHECK-NEXT: has_rank: true // CHECK-NEXT: } ], // CHECK-NEXT: inputs: [ ], // CHECK-NEXT: outputs: [ 0 ],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 14 16:41:28 UTC 2022 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/isolate-placer.mlir
func.func @main() { tf_executor.graph { %0:2 = tf_executor.island wraps "tf.VarHandleOp"() {container = "c", shared_name = "n"} : () -> tensor<!tf_type.resource<tensor<8xf32>>> %1:2 = tf_executor.island wraps "tf.StatefulPartitionedCall"(%0#0) {Tin = ["tfdtype$DT_RESOURCE"], Tout = ["tfdtype$DT_RESOURCE"], config = "", config_proto = "", executor_type = "", f = @foo} : (tensor<!tf_type.resource<tensor<8xf32>>>) -> tensor<!tf_type.resource<tensor<8xf32>>> loc("call_foo")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/signature_def_output_override.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 14 16:41:28 UTC 2022 - 4.5K bytes - Viewed (0)