Home
last modified time | relevance | path

Searched refs:function_def (Results 1 – 25 of 73) sorted by relevance

123

/external/tensorflow/tensorflow/core/data/
Drewrite_utils.cc80 void AddFakeSinks(FunctionDef* function_def) { in AddFakeSinks() argument
82 for (const auto& output : function_def->signature().output_arg()) { in AddFakeSinks()
83 NodeDef* node = function_def->add_node_def(); in AddFakeSinks()
85 strings::StrCat("FakeSink", counter++), function_def, node); in AddFakeSinks()
87 node->add_input(function_def->ret().at(output.name())); in AddFakeSinks()
90 (*function_def->mutable_ret())[output.name()] = in AddFakeSinks()
95 void RemoveFakeSinks(FunctionDef* function_def) { in RemoveFakeSinks() argument
98 for (const auto& node : function_def->node_def()) { in RemoveFakeSinks()
103 for (const auto& output_arg : function_def->signature().output_arg()) { in RemoveFakeSinks()
104 const std::string& tensor = function_def->ret().at(output_arg.name()); in RemoveFakeSinks()
[all …]
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dsaved_model_utils.cc133 const FunctionDef* function_def) { in ValidateSavedFunctionCompatibleWithFunctionDef() argument
155 const std::string& name = function_def->signature().name(); in ValidateSavedFunctionCompatibleWithFunctionDef()
162 function_def->signature().input_arg_size()) { in ValidateSavedFunctionCompatibleWithFunctionDef()
165 function_def->signature().input_arg_size(), in ValidateSavedFunctionCompatibleWithFunctionDef()
175 if (output_specs.size() != function_def->signature().output_arg_size()) { in ValidateSavedFunctionCompatibleWithFunctionDef()
178 function_def->signature().output_arg_size(), in ValidateSavedFunctionCompatibleWithFunctionDef()
254 const FunctionDef* function_def, in LoadTFConcreteFunction() argument
259 saved_concrete_function, function_def)); in LoadTFConcreteFunction()
274 return TFConcreteFunction::Create(function_def, std::move(captures), {}, ctx, in LoadTFConcreteFunction()
377 for (const FunctionDef& function_def : library.function()) { in FunctionNameToFunctionDefMap() local
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
Dtpu_test.cc34 FunctionDef* function_def = library->add_function(); in TEST_F() local
35 function_def->add_node_def()->set_op("Mul"); in TEST_F()
55 FunctionDef* function_def = library->add_function(); in TEST_F() local
56 function_def->add_node_def()->set_op("TPUPartitionedCall"); in TEST_F()
Dtpu.cc31 for (const auto& function_def : def.library().function()) { in IsLegacyTPUBridgeGraphDef() local
32 for (const auto& node : function_def.node_def()) { in IsLegacyTPUBridgeGraphDef()
/external/tensorflow/tensorflow/core/function/
Druntime_client_test.py118 fndef = rt.GetFunctionProto(cf.function_def.signature.name)
120 self.assertEqual(fndef.signature.name, cf.function_def.signature.name)
134 fndef = rt.GetFunctionProto(cf.function_def.signature.name)
155 fndef = rt.GetFunctionProto(cf.function_def.signature.name)
175 fndef = rt.GetFunctionProto(cf.function_def.signature.name)
196 rt.TransformFunction(cf.function_def.signature.name, "test-pass")
Druntime_client.py34 def CreateFunction(self, function_def: function_pb2.FunctionDef):
35 self.CreateFunctionFromString(function_def.SerializeToString())
/external/tensorflow/tensorflow/core/grappler/optimizers/inference/
Dbatch_op_rewriter.cc165 FunctionDef* function_def = in Optimize() local
167 for (int j = 0; j < function_def->node_def_size(); j++) { in Optimize()
168 NodeDef* node = function_def->mutable_node_def(j); in Optimize()
196 FunctionDef* function_def = in Optimize() local
198 for (int j = 0; j < function_def->node_def_size(); j++) { in Optimize()
199 NodeDef* node = function_def->mutable_node_def(j); in Optimize()
Dbatch_op_rewriter_test.cc77 FunctionDef* function_def = function_def_lib->add_function(); local
78 set_batch_node_attribute(num_batch_threads, function_def->add_node_def());
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_api_info.cc29 Status FunctionApiInfo::Init(const FunctionDef& function_def) { in Init() argument
31 for (const auto& attr : function_def.attr()) { in Init()
48 input_arg_dtypes_.reserve(function_def.signature().input_arg_size()); in Init()
49 for (const auto& input_arg : function_def.signature().input_arg()) { in Init()
52 output_arg_dtypes_.reserve(function_def.signature().output_arg_size()); in Init()
53 for (const auto& output_arg : function_def.signature().output_arg()) { in Init()
59 "Function '", function_def.signature().name(), in Init()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation_test.cc58 tensorflow::FunctionDef function_def; in TEST() local
63 &function_def)); in TEST()
64 TF_ASSERT_OK(ctx->AddFunctionDef(function_def)); in TEST()
Dkernel_and_device.cc143 const FunctionDef* function_def; in InstantiateFunc() local
148 function_def = pflr_->GetFLR(host_cpu_device_->name()) in InstantiateFunc()
152 function_def = flr_->GetFunctionLibraryDefinition()->Find(ndef.op()); in InstantiateFunc()
155 if (function_def != nullptr) { in InstantiateFunc()
156 op_def = &(function_def->signature()); in InstantiateFunc()
206 options.config_proto, function_def->signature().name(), in InstantiateFunc()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dflat_tensor_function.cc49 const FunctionDef* function_def, in Create() argument
52 TF_RETURN_IF_ERROR(ctx->AddFunctionDef(*function_def)); in Create()
60 out->reset(new FlatTensorFunction(function_def->signature().name(), in Create()
Dtf_concrete_function.cc41 const FunctionDef* function_def, in Create() argument
47 function_def, std::move(captures), ctx, &func)); in Create()
Dtf_signature_def_function.cc42 const FunctionDef* function_def, in Create() argument
48 function_def, std::move(captures), ctx, &func)); in Create()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmake_deterministic.cc219 const FunctionDef& function_def, in GetAllTransitiveDependencies() argument
225 NameToNode(function_def); in GetAllTransitiveDependencies()
265 const FunctionDef* function_def = library.Find(func.name()); in SplitMap() local
266 if (!function_def) { in SplitMap()
272 GetAllTransitiveDependencies(*function_def, nondeterministic_nodes); in SplitMap()
282 split_utils::SplitFunction(*function_def, nodes_to_move, in SplitMap()
525 const FunctionDef* function_def = library.Find(function_name); in FunctionMayIntroduceNondeterminism() local
526 if (!function_def) { in FunctionMayIntroduceNondeterminism()
532 for (const NodeDef& node_def : function_def->node_def()) { in FunctionMayIntroduceNondeterminism()
655 const FunctionDef* function_def = library.Find(function_name); in GraphMayHaveAsyncNondeterminism() local
[all …]
Dfunction_utils.cc184 const FunctionDef& function_def, bool skip_assert) { in IsFunctionStateful() argument
185 if (!function_def.signature().is_stateful()) return false; in IsFunctionStateful()
187 for (const NodeDef& node_def : function_def.node_def()) { in IsFunctionStateful()
/external/tensorflow/tensorflow/c/eager/
Dc_api_distributed_test.cc191 const string function_def = AddVariablesFunction(); in TestFunctionWithPackedInput() local
192 TFE_ContextAddFunctionDef(ctx, function_def.data(), function_def.size(), in TestFunctionWithPackedInput()
382 const string function_def = VariableAddFunction(); in TEST() local
383 TFE_ContextAddFunctionDef(ctx, function_def.data(), function_def.size(), in TEST()
505 const string function_def = inject_error ? VariableAddFunctionWithGraphError() in TestDistributedFunctionCancellation() local
507 TFE_ContextAddFunctionDef(ctx, function_def.data(), function_def.size(), in TestDistributedFunctionCancellation()
Dc_api_remote_test_util.cc140 string function_def = MatMulFunction(matmul_device); in TestRemoteExecuteSilentCopies() local
141 TFE_ContextAddFunctionDef(ctx, function_def.data(), function_def.size(), in TestRemoteExecuteSilentCopies()
/external/python/cpython3/Parser/
Dpegen.c2244 _PyPegen_function_def_decorators(Parser *p, asdl_expr_seq *decorators, stmt_ty function_def) in _PyPegen_function_def_decorators() argument
2246 assert(function_def != NULL); in _PyPegen_function_def_decorators()
2247 if (function_def->kind == AsyncFunctionDef_kind) { in _PyPegen_function_def_decorators()
2249 function_def->v.FunctionDef.name, function_def->v.FunctionDef.args, in _PyPegen_function_def_decorators()
2250 function_def->v.FunctionDef.body, decorators, function_def->v.FunctionDef.returns, in _PyPegen_function_def_decorators()
2251 function_def->v.FunctionDef.type_comment, function_def->lineno, in _PyPegen_function_def_decorators()
2252 function_def->col_offset, function_def->end_lineno, function_def->end_col_offset, in _PyPegen_function_def_decorators()
2257 function_def->v.FunctionDef.name, function_def->v.FunctionDef.args, in _PyPegen_function_def_decorators()
2258 function_def->v.FunctionDef.body, decorators, in _PyPegen_function_def_decorators()
2259 function_def->v.FunctionDef.returns, in _PyPegen_function_def_decorators()
[all …]
/external/tensorflow/tensorflow/python/eager/
Dcontext_test.py115 function_def = context.get_function_def(concrete.name)
117 self.assertIsNot(function_def, None)
120 for node_def in function_def.node_def:
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc157 Status ShapeRefiner::InferShapesForFunction(const FunctionDef* function_def, in InferShapesForFunction() argument
161 auto it = functions_.find(function_def); in InferShapesForFunction()
167 *function_def, attributes, in InferShapesForFunction()
178 functions_[function_def].reset(new_graph); in InferShapesForFunction()
697 const FunctionDef* function_def = in RunShapeFn() local
699 if (function_def != nullptr) { in RunShapeFn()
706 function_def, AttrSlice(&function.attr()), c); in RunShapeFn()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device_util.cc764 const FunctionDef* function_def = doperation.function_def; in PrepareGraphForMlir() local
765 if (function_def) { in PrepareGraphForMlir()
776 for (const auto& out : function_def->signature().output_arg()) in PrepareGraphForMlir()
942 Status MaybeInsertIdentityNodes(const FunctionDef* function_def, Graph* graph) { in MaybeInsertIdentityNodes() argument
943 if (function_def == nullptr || function_def->control_ret().empty()) { in MaybeInsertIdentityNodes()
976 void AddDTensorFunctionAttr(FunctionDef& function_def) { in AddDTensorFunctionAttr() argument
981 function_def.mutable_attr()->insert( in AddDTensorFunctionAttr()
989 function_def.mutable_attr()->insert( in AddDTensorFunctionAttr()
Ddtensor_device_util.h129 const FunctionDef* function_def; member
131 inline bool is_func() const { return function_def != nullptr; } in is_func()
620 Status MaybeInsertIdentityNodes(const FunctionDef* function_def, Graph* graph);
623 void AddDTensorFunctionAttr(FunctionDef& function_def);
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py795 def _find_children_hints_in_while_loop(function_def, nodes_mapping): argument
808 for node in function_def.node_def:
813 name_to_seq_num = _extract_topology_sequence_mapping(function_def.node_def)
864 for function_def in graph_def.library.function:
865 if function_def.signature.name == body_name:
866 function_inputs = function_def.signature.input_arg
873 function_def, nodes_mapping)
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util_test.cc291 FunctionDef* function_def = graph_def.mutable_library()->add_function(); in TEST() local
292 function_def->mutable_signature()->set_name("F"); in TEST()
294 function_def->add_node_def()->set_op(op); in TEST()

123