| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 628 const FunctionDef* func_def = flib.Find(func_name); in PropagateTFDataAttrs() local
630 if (func_def == nullptr) return; in PropagateTFDataAttrs()
640 for (const NodeDef& node : func_def->node_def()) { in PropagateTFDataAttrs()
660 for (const auto& func_def : fdef_lib.function()) { in PropagateTFDataAttrs() local
661 const std::string& func_name = func_def.signature().name(); in PropagateTFDataAttrs()
662 if (data::IsTFDataFunction(func_def)) in PropagateTFDataAttrs()
668 for (FunctionDef& func_def : *fdef_lib.mutable_function()) { in PropagateTFDataAttrs()
669 const std::string& func_name = func_def.signature().name(); in PropagateTFDataAttrs()
671 !data::IsTFDataFunction(func_def)) { in PropagateTFDataAttrs()
673 (*func_def.mutable_attr())[data::kTFDataFunction].set_b(true); in PropagateTFDataAttrs()
[all …]
|
| D | function_api_info_test.cc | 53 FunctionDef* func_def) { in PopulateFunction() argument
54 OpDef* sig = func_def->mutable_signature(); in PopulateFunction()
59 auto* func_attr = func_def->mutable_attr(); in PopulateFunction()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_def_util.cc | 131 for (const auto& func_def : graph_def.library().function()) { in IsFunction() local
132 if (op_name == func_def.signature().name()) return true; in IsFunction()
152 FunctionDef* func_def = graph_def->mutable_library()->mutable_function(f); in RemoveNewDefaultAttrsFromGraphDef() local
153 for (int n = 0; n < func_def->node_def_size(); ++n) { in RemoveNewDefaultAttrsFromGraphDef()
154 NodeDef* node_def = func_def->mutable_node_def(n); in RemoveNewDefaultAttrsFromGraphDef()
|
| D | function_test.cc | 1363 auto func_def = FDH::Create( in TEST() local
1370 auto* attr = func_def.mutable_attr(); in TEST()
1373 return func_def; in TEST()
|
| D | function.cc | 813 string DebugString(const FunctionDef& func_def) { return Print(func_def); } in DebugString() argument
1639 const auto& func_def = flib.Find(func_name); in ReachableFunctions() local
1640 const auto attr_it = func_def->attr().find(kApiImplements); in ReachableFunctions()
1641 if (attr_it != func_def->attr().end() && in ReachableFunctions()
|
| D | function.h | 254 std::string DebugString(const FunctionDef& func_def);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 634 FunctionDef func_def; in ConvertLibFunction() local
636 GraphToFunctionDef(*sub_graph, function_name, control_ret, &func_def)); in ConvertLibFunction()
642 for (auto& node_def : *func_def.mutable_node_def()) { in ConvertLibFunction()
665 func_def.mutable_signature()->set_is_stateful(true); in ConvertLibFunction()
676 func_def.mutable_attr())); in ConvertLibFunction()
682 (*func_def.mutable_resource_arg_unique_id())[i] = in ConvertLibFunction()
696 (*func_def.mutable_arg_attr())[i] = std::move(func_def_arg_i_attrs); in ConvertLibFunction()
699 (*flib->add_function()) = std::move(func_def); in ConvertLibFunction()
732 for (auto& func_def : flib.function()) { in Convert() local
733 TF_RETURN_IF_ERROR(flib_def->AddFunctionDef(func_def)); in Convert()
[all …]
|
| D | import_model.cc | 1241 const auto* func_def = func_lib.Find(std::string(func_name)); in ConvertLibFunction() local
1242 if (func_def == nullptr) { in ConvertLibFunction()
1251 FunctionDefToBodyHelper(*func_def, AttrSlice(), &func_lib, &fbody)); in ConvertLibFunction()
1255 attributes.reserve(func_def->attr_size()); in ConvertLibFunction()
1256 for (const auto& name_and_value : func_def->attr()) { in ConvertLibFunction()
1267 if (func_def->signature().is_stateful()) { in ConvertLibFunction()
1290 for (const auto& name_and_value : func_def->attr()) { in ConvertLibFunction()
1293 auto& signature = func_def->signature(); in ConvertLibFunction()
1864 const auto* func_def = graph_flib_.Find(node_type_name); in ConvertNode() local
1866 if (func_def) { in ConvertNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | meta_optimizer.cc | 136 for (const FunctionDef& func_def : in Optimize() local
138 if (flib.Find(func_def.signature().name()) == nullptr) { in Optimize()
139 TF_RETURN_IF_ERROR(flib.AddFunctionDef(func_def)); in Optimize()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | tfr_decompose_ctx.cc | 174 FunctionDef func_def; in ExpandNode() local
176 ConvertMlirFunctionToFunctionLibraryDef(func, export_confs_, &func_def)); in ExpandNode()
178 return func_def; in ExpandNode()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime.cc | 43 const FunctionDef* func_def = flib_def.Find(sig.name()); in ConstructFunctionGraph() local
44 if (func_def == nullptr) { in ConstructFunctionGraph()
52 flib_def.ReachableDefinitions(*func_def); in ConstructFunctionGraph()
160 TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(*func_def, attrs, &pruned_flib_def, in ConstructFunctionGraph()
|
| /external/sl4a/Docs/ |
| D | generate_api_reference_md.py | 49 def __init__(self, rpc_def, func_def): argument
57 self._ParseFunctionDefinition(func_def)
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass.cc | 349 const FunctionDef* func_def = flib_def.Find(function_name); in InferDeviceForCluster() local
350 TF_RET_CHECK(func_def) << "Could not find " << function_name; in InferDeviceForCluster()
354 for (const NodeDef& ndef : func_def->node_def()) { in InferDeviceForCluster()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.h | 120 bool GetFunctionDef(TF_Function* func, tensorflow::FunctionDef* func_def);
|
| D | c_test_util.cc | 412 bool GetFunctionDef(TF_Function* func, tensorflow::FunctionDef* func_def) { in GetFunctionDef() argument
418 if (ret) ret = func_def->ParseFromArray(buffer->data, buffer->length); in GetFunctionDef()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen.py | 233 def lookup(self, f_name, func_def=None, optional=False): argument
237 if isinstance(func_def, types.FunctionType):
238 if not hasattr(func_def, '_tfr_op_name'):
244 op_name = getattr(func_def, '_tfr_op_name')
245 elif not func_def:
250 for dec in func_def.decorator_list:
|
| /external/python/mock/mock/tests/ |
| D | testhelpers.py | 423 func_def = "def foo(a, *, b=None): pass\n"
425 exec (func_def, namespace)
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 2622 const FunctionDef* func_def = fld.Find(func_name); in ContainsResourceWriteOp() local
2623 for (const NodeDef& n : func_def->node_def()) { in ContainsResourceWriteOp()
|