- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 91 for PartitionedCall (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_sharding_identification.mlir
%1:2 = "tf.StatefulPartitionedCall"(%arg0){f= @func_body, config="", config_proto="", executor_type=""} : (tensor<*xi32>) -> (tensor<*xi32>, tensor<*xi1>) %2 = "tf.PartitionedCall"(%arg1) {config = "", config_proto = "", executor_type = "", f = @pcall_func_body} : (tensor<*xi1>) -> (tensor<i32>) %3, %4 = "tf.A"(%1#0, %2) : (tensor<*xi32>, tensor<i32>) -> (tensor<*xi32>, tensor<*xi1>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 20 19:07:52 UTC 2024 - 47.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
} } if_op.replaceAllUsesWith(new_if); if_op.erase(); return success(); } // Handles stack usage by a tf.StatefulPartitionedCall or a tf.PartitionedCall. // It will first check if the callee was previously handled, and try to reuse // that result if so. Otherwise, it will clone and convert the callee function, // and performs stack ops decomposition on it.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/promote_resources_to_args.mlir
%1 = "tf._SomeOp"() : () -> tensor<2xf32> "tf.AssignVariableOp"(%0, %1) : (tensor<!tf_type.resource<tensor<2xf32>>>, tensor<2xf32>) -> () %2 = "tf.PartitionedCall"(%0) {config = "", config_proto = "", executor_type = "", f = @callee} : (tensor<!tf_type.resource<tensor<2xf32>>>) -> tensor<2xf32> func.return %2 : tensor<2xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 18.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td
MHLO_Tensor:$on_false), (CHLO_BroadcastSelectOp $pred, $on_true, $on_false)>; //===----------------------------------------------------------------------===// // PartitionedCall and LegacyCall op patterns. //===----------------------------------------------------------------------===// def ArgTypesMatchCallee : Constraint<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 122.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir
// CHECK-NEXT: [[results_1:%.*]] = call @callee( // CHECK-SAME: (!tf_mlrt.tensor) -> !tf_mlrt.tensor %2 = "tf.PartitionedCall"(%0) {config = "", config_proto = "", executor_type = "", f = @callee} : (tensor<i32>) -> (tensor<i32>) // CHECK-NEXT: [[results_2:%.*]] = call @callee( // CHECK-SAME: (!tf_mlrt.tensor) -> !tf_mlrt.tensor
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
``` `_XlaMustCompile=true` in the following code ```mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
``` `_XlaMustCompile=true` in the following code ```mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
//===--------------------------------------------------------------------===// // tf.{|Stateful}PartitionedCall //===--------------------------------------------------------------------===// // Test valid tf.PartitionedCall // CHECK-LABEL: func @testValidPartitionedCall func.func @testValidPartitionedCall(%arg0: tensor<i32>) -> tensor<i32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0)