- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 68 for StatefulPartitionedCall (0.61 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/xla_validate_inputs.mlir
func.func @nested_entry_functions() attributes {tf.entry_function = {}} { tf_executor.graph { %control = tf_executor.island wraps "tf.StatefulPartitionedCall"() {config = "", config_proto = "", device = "/device:CPU:0", executor_type = "", f = @func} : () -> () tf_executor.fetch } func.return } func.func @func() attributes {tf.entry_function = {}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 19:29:14 UTC 2024 - 818 bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_initializer_function_ops_to_main.mlir
func.func @main(%arg0: tensor<?x!tf_type.string> {tf_saved_model.index_path = ["serving_default_input_vocabs:0"]}) -> (tensor<*xi64> {tf_saved_model.index_path = ["StatefulPartitionedCall:0"]}) attributes {tf.entry_function = {inputs = "serving_default_input_vocabs:0", outputs = "StatefulPartitionedCall:0"}, tf_saved_model.exported_names = ["main"]} { %0 = tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:49:35 UTC 2023 - 29.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/optimize-arg-operand-constraint.mlir
%0 = arith.constant dense<2.000000e+00> : tensor<f32> %1 = arith.constant dense<1.000000e+00> : tensor<f32> %2 = "tf.AddV2"(%arg0, %1) {T = "tfdtype$DT_FLOAT", device = "", name = "StatefulPartitionedCall/add"} : (tensor<1xf32>, tensor<f32>) -> tensor<1xf32> %3 = "tf.Mul"(%2, %0) {T = "tfdtype$DT_FLOAT", device = "", name = "output_node"} : (tensor<1xf32>, tensor<f32>) -> tensor<1xf32> func.return %3 : tensor<1xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:47:26 UTC 2022 - 719 bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); Node* call; NameAttrList f_name_attr; f_name_attr.set_name(fd.signature().name()); TF_ASSERT_OK( NodeBuilder("B", "StatefulPartitionedCall", &root.graph()->flib_def()) .Input(inputs) .Attr("Tin", {DT_RESOURCE}) .Attr("Tout", {DT_RESOURCE}) .Attr("f", f_name_attr) .Finalize(root.graph(), &call));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/resource_inlining.mlir
// CHECK-NOT: "tf.Cast" %0 = "tf.VarHandleOp"() {_class = ["loc:@Variable"], allowed_devices = [], container = "", device = "", shared_name = "Variable"} : () -> tensor<!tf_type.resource<tensor<f32>>> %1 = "tf.StatefulPartitionedCall"(%0) {config = "", config_proto = "", executor_type = "", f = @callee} : (tensor<!tf_type.resource<tensor<f32>>>) -> tensor<f32> func.return %1 : tensor<f32> } // CHECK-NOT: func private @callee
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:47:26 UTC 2022 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
::llvm::cl::desc("Skip tf.ResizeBilinear and tf.ResizeNearestNeighbor")}; Option<bool> skip_partitioned_calls_{ *this, "skip-partitioned-calls", ::llvm::cl::desc( "Skip tf.StatefulPartitionedCall and tf.PartitionedCall")}; }; void TFToMhloPass::runOnOperation() { auto func = getOperation(); MLIRContext *context = func->getContext(); RewritePatternSet patterns(context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0) -
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/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)