- Sort Score
- Result 10 results
- Languages All
Results 101 - 110 of 116 for PartitionedCall (0.23 sec)
-
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) -
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) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// `potential_refined_type`. Return true if the type was changed. bool RefineResultType(Operation* op, Value result, Type potential_refined_type); // Infers the shape from a (Stateful)PartitionedCall operation by looking up // the called function and propagating the return type. bool InferShapeForCall(CallOpInterface call_op); bool InferShapeForCast(Operation* op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)