- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 71 for stateful_partitioned_call (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_freeze_global_tensors.mlir
func.func @f(%arg0: tensor<!tf_type.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) attributes {tf_saved_model.exported_names = ["f"]} { "tf.StatefulPartitionedCall"(%arg0) {config = "", config_proto = "", executor_type = "", f = @f_callee} : (tensor<!tf_type.resource<tensor<f32>>>) -> () func.return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 03:07:35 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-control-ret-diff-island.pbtxt
key: "value" value { tensor { dtype: DT_FLOAT tensor_shape { } float_val: 0.0 } } } } node { name: "test_fn_call" op: "StatefulPartitionedCall" input: "const" attr { key: "Tin" value { list { type: DT_FLOAT } } } attr { key: "Tout" value { list { type: DT_FLOAT
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 10 19:53:21 UTC 2020 - 1.9K bytes - Viewed (0) -
tensorflow/cc/gradients/functional_grad.cc
for (int i = 0; i < num_inputs; i++) { grad_outputs->push_back(grad[i]); } return scope.status(); } REGISTER_GRADIENT_OP("PartitionedCall", PartitionedCallGrad); REGISTER_GRADIENT_OP("StatefulPartitionedCall", PartitionedCallGrad); } // anonymous namespace } // namespace ops
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 15 20:09:06 UTC 2021 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/sink_variable_as_named_array.mlir
// // CHECK-LABEL: func.func @serving_default // CHECK-NOT: IfrtLoadVariable // CHECK: "tf.VarHandleOp" // CHECK-NEXT: "tf.AssignVariableOp" // CHECK-NEXT: "tf.ReadVariableOp" // CHECK-NEXT: "tf.StatefulPartitionedCall" // CHECK-NEXT: return // module { func.func @serving_default() -> tensor<*xi32> { %cst = "tf.Const"() <{value = dense<"some_test.txt"> : tensor<!tf_type.string>}> : () -> tensor<!tf_type.string>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 15:33:17 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
NodeWith(Op("Switch"), Inputs(Out(0, xla_compile), Out(1, xla_compile))); auto xla_run = NodeWith(Op("_XlaRun"), Inputs(Out(1, predicated_compilation_key))); auto tf_call = NodeWith(Op("StatefulPartitionedCall"), CtrlDeps(NodeWith(Op("Identity"), Inputs(Out(0, predicated_compilation_key))))); auto merge = NodeWith(Op("_XlaMerge"), Inputs(Out(tf_call), Out(xla_run)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-control-ret-same-island.pbtxt
key: "value" value { tensor { dtype: DT_FLOAT tensor_shape { } float_val: 0.0 } } } } node { name: "test_fn_call" op: "StatefulPartitionedCall" input: "const" attr { key: "Tin" value { list { type: DT_FLOAT } } } attr { key: "Tout" value { list { type: DT_FLOAT
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 10 19:53:21 UTC 2020 - 1.8K bytes - Viewed (0) -
tensorflow/cc/gradients/functional_grad_test.cc
(*f.mutable_attr())["T"].set_type(DT_FLOAT); auto results = PartitionedCall(scope_, std::initializer_list<Input>{x}, {DT_FLOAT}, f); RunTest(x, {}, results[0], {}); auto stateful_results = StatefulPartitionedCall( scope_, std::initializer_list<Input>{x}, {DT_FLOAT}, f); RunTest(x, {}, stateful_results[0], {}); } } // namespace } // namespace ops
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 26 14:42:24 UTC 2021 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/stack_ops_decomposition.mlir
// ----- // Tests PartitionedCall/StatefulPartitionedCall. // CHECK-LABEL: func @main func.func @main(%arg0: tensor<i1>) -> () { %max_size = "tf.Const"() {value = dense<10> : tensor<i32>} : () -> tensor<i32> // CHECK-NOT: tf.Stack %stack = "tf.StackV2"(%max_size) {elem_type = f32, stack_name = "s"} : (tensor<i32>) -> tensor<!tf_type.resource> // CHECK: "tf.StatefulPartitionedCall" // CHECK-SAME: f = @callee_stack_decomposed
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 25.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/sink_in_invariant_ops.mlir
// CHECK: [[handle:%.*]] = "tf.VarHandleOp"() %handle = "tf.VarHandleOp"() {container = "", shared_name = "x"} : () -> tensor<!tf_type.resource<tensor<i32>>> // CHECK: "tf.StatefulPartitionedCall"([[handle]]) %x = "tf.StatefulPartitionedCall"(%handle) {device = "/CPU:0", config = "", config_proto = "", executor_type = "", f = @some_func} : (tensor<!tf_type.resource<tensor<i32>>>) -> (tensor<i32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/fetch_feed_names.mlir
%0 = tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 3K bytes - Viewed (0)