- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for mutable_signature (0.14 sec)
-
tensorflow/c/experimental/saved_model/core/tf_concrete_function_loading_test.cc
testing::CreateTensorHandle(ctx, DT_FLOAT, {2, 4}, value)) {} }; FunctionDef FuncDefWithNumInputsOutputs(int num_inputs, int num_outputs) { FunctionDef func; OpDef* signature = func.mutable_signature(); for (int i = 0; i < num_inputs; ++i) { signature->add_input_arg(); } for (int i = 0; i < num_outputs; ++i) { signature->add_output_arg(); } return func; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 19:16:58 UTC 2023 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
&rewritten_fdef)); const auto new_function_name = fld->UniqueFunctionName(absl::StrCat(n->type_string(), "_lifted_arg_")); rewritten_fdef.mutable_signature()->set_name(new_function_name); TF_RETURN_IF_ERROR(fld->AddFunctionDef(rewritten_fdef)); // We need to recreate the node. Otherwise TF will not know n->num_inputs() // has increased.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
} auto stateful_string = mlir::TF::TensorFlowDialect::GetStatefulAttrName(); if (auto attr = function->getAttrOfType<mlir::UnitAttr>(stateful_string)) { func_def.mutable_signature()->set_is_stateful(true); } // Ignore the gradient and is_stateful attribute on the function as they have // been handled above. Ignore the entry func attribute as it is an MLIR
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
} auto stateful_string = mlir::TF::TensorFlowDialect::GetStatefulAttrName(); if (auto attr = function->getAttrOfType<mlir::UnitAttr>(stateful_string)) { func_def.mutable_signature()->set_is_stateful(true); } // Ignore the gradient and is_stateful attribute on the function as they have // been handled above. Ignore the entry func attribute as it is an MLIR
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
FunctionDefHelper::Define("UncompilableFn", {"n_a:float"}, {"n_c:float"}, {}, {{{"n_c"}, "UncompilableUnary", {"n_a"}}}); FunctionDef noinline = compilable; noinline.mutable_signature()->set_name("NoInlineFn"); AddAttr("_noinline", static_cast<bool>(true), noinline.mutable_attr()); FunctionDefLibrary flib; *flib.add_function() = compilable; *flib.add_function() = uncompilable;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0)