- Sort Score
- Result 10 results
- Languages All
Results 41 - 44 of 44 for PartitionedCall (0.21 sec)
-
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/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/tensor_list_ops_decomposition.cc
arg_buffer_size_is_fixed(i)}; } } UpdateFuncType(func); } // Holds information about a decomposed callee function for // PartitionedCall/StatefulPartitionedCall. struct PartitionedCallDecompositionInfo { bool signature_change; func::FuncOp decomposed_callee; llvm::SmallDenseMap<int64_t, int64_t> buffer_arg_to_size_arg;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0)