- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for mutable_signature (0.13 sec)
-
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)