- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 56 for PartitionedCall (0.19 sec)
-
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/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/jit/mark_for_compilation_pass_test.cc
NameAttrList b_name_attr; b_name_attr.set_name("CompilableFn"); ops::PartitionedCall b(root.WithOpName("B"), {a, a}, {DT_FLOAT}, b_name_attr); NameAttrList c_name_attr; c_name_attr.set_name("UncompilableFn"); ops::PartitionedCall c(root.WithOpName("C"), {a}, {DT_FLOAT}, c_name_attr); Output d = ops::Add(root.WithOpName("D"), b.output.front(), c.output.front());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K 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) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/while_to_map_fn.mlir
func.func @serving_default(%arg0: tensor<?xf32> {tf.device = "/job:localhost/replica:0/task:0/device:CPU:0"}) -> tensor<3xf32> attributes {tf.entry_function = {control_outputs = "", inputs = "serving_default_input:0", outputs = "PartitionedCall:0"}} { %cst = "tf.Const"() {device = "/job:localhost/replica:0/task:0/device:CPU:0", value = dense<0> : tensor<i32>} : () -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:40:22 UTC 2024 - 68.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
%3 = arith.constant dense<0> : tensor<i32> // CHECK: PartitionedCall // CHECK-SAME: f = @add // CHECK-SAME: device = "noodle" %4 = "tf.Case"(%2, %arg0, %arg1) {branches = [@sub, @add], output_shapes = [#tf_type.shape<>], device = "noodle", is_stateless = false} : (tensor<i32>, tensor<f32>, tensor<f32>) -> tensor<f32> // CHECK: PartitionedCall // CHECK-SAME: f = @sub // CHECK-SAME: _cluster_launch = "not_ready"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0)