Searched refs:fdef (Results 1 - 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/jit/
H A Dgraph_to_functiondef.h25 // Converts 'graph' to a FunctionDef 'fdef', with name 'name'.
29 FunctionDef* fdef);
H A Dgraph_to_functiondef_test.cc63 FunctionDef fdef; local
64 TF_EXPECT_OK(GraphToFunctionDef(*graph, "test_fn", &fdef));
81 bool fdefs_equal = EqualFunctionDef(fdef_expected, fdef, &diff);
99 FunctionDef fdef; local
100 TF_EXPECT_OK(GraphToFunctionDef(*graph, "test_fn", &fdef));
114 bool fdefs_equal = EqualFunctionDef(fdef_expected, fdef, &diff);
H A Dgraph_to_functiondef.cc116 FunctionDef* fdef) {
117 fdef->mutable_signature()->set_name(name);
129 while (fdef->signature().input_arg_size() <= index) {
130 fdef->mutable_signature()->add_input_arg();
133 fdef->mutable_signature()->mutable_input_arg(index);
146 while (fdef->signature().output_arg_size() <= index) {
147 fdef->mutable_signature()->add_output_arg();
150 fdef->mutable_signature()->mutable_output_arg(index);
161 NodeDef* node_def = fdef->add_node_def();
219 for (int i = 0; i < fdef
115 GraphToFunctionDef(const Graph& graph, const string& name, FunctionDef* fdef) argument
[all...]
H A Dcreate_xla_launch_op.cc66 DebugString(fbody->fdef));
94 &fbody->fdef.signature(), flr, fbody->arg_types, input_memory_types,
H A Dmark_for_compilation_pass.cc126 const FunctionDef& fdef = fbody->fdef; local
128 if (GetNodeAttr(AttrSlice(&fdef.attr()), "_noinline", &noinline).ok() &&
135 VLOG(2) << "Can't compile noinline function: " << fdef.DebugString();
H A Dencapsulate_subgraphs_pass.cc909 FunctionDef fdef; local
910 TF_RETURN_IF_ERROR(GraphToFunctionDef(*graph_, name, &fdef));
917 strings::StrCat("encapsulate_fdef_", name), fdef);
921 TF_RETURN_IF_ERROR(library->AddFunctionDef(fdef));
966 FunctionDef fdef; local
967 TF_RETURN_IF_ERROR(GraphToFunctionDef(*graph_, name, &fdef));
975 strings::StrCat("replace_encapsulate_fdef_", name), fdef);
979 TF_RETURN_IF_ERROR(library->AddFunctionDef(fdef));
1843 const FunctionDef* fdef = library->Find(node->type_string()); local
1844 if (fdef
[all...]
/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dtest_util.cc26 const FunctionDef* fdef = library.Find(name); local
27 TF_RET_CHECK(fdef != nullptr);
34 InstantiateFunction(*fdef, AttrSlice(), get_func_sig, &inst));
H A Ddump_graph.h45 string DumpFunctionDefToFile(const string& name, FunctionDef const& fdef);
H A Ddump_graph.cc85 string DumpFunctionDefToFile(const string& name, FunctionDef const& fdef) { argument
87 Status status = WriteTextProto(Env::Default(), path, fdef);
/external/tensorflow/tensorflow/core/framework/
H A Dfunction_test.cc67 auto fdef = FDH::Create( local
95 EXPECT_EQ(DebugString(fdef), e);
100 InstantiateFunction(fdef, Attrs({{"T", DT_FLOAT}}), GetOpSig, &result));
114 auto fdef = FDH::Create( local
141 EXPECT_EQ(DebugString(fdef), e);
146 InstantiateFunction(fdef, Attrs({{"T", DT_FLOAT}}), GetOpSig, &result));
167 auto fdef = FDH::Create( local
188 EXPECT_EQ(DebugString(fdef), e);
191 TF_ASSERT_OK(InstantiateFunction(fdef, AttrSlice(), GetOpSig, &result));
204 auto fdef local
258 auto fdef = FDH::Create( local
305 auto fdef = FDH::Define( local
374 auto fdef = FDH::Create( local
447 auto fdef = FDH::Define( local
504 auto fdef = local
524 auto fdef = local
533 auto fdef = FDH::Define("test", {}, {}, {"T:{float, double, int32, int64}"}, local
544 auto fdef = FDH::Define("test", {"x:float", "x:float"}, {}, {}, {}); local
551 auto fdef = FDH::Define("test", {"x:float"}, {}, {}, local
562 auto fdef = FDH::Create("test", {"x:float"}, {}, {}, local
573 auto fdef = FDH::Define("test", {"x:float"}, {}, {}, local
583 auto fdef = local
594 auto fdef = FDH::Create("test", {}, {"y: float"}, {}, local
605 auto fdef = FDH::Create("test", {}, {"y: float"}, {}, local
616 auto fdef = FDH::Create("test", {}, {"y: float"}, {}, local
639 auto fdef = FDH::Define("test", {}, {"y: float"}, {}, local
649 auto fdef = FDH::Create( local
675 auto fdef = FDH::Create( local
702 auto fdef = FDH::Create( local
729 auto fdef = FDH::Create( local
750 auto fdef = FDH::Create( local
771 auto fdef = FDH::Create( local
799 auto fdef = FDH::Create( local
827 auto fdef = FDH::Create( local
973 FunctionDef fdef = test::function::XTimesTwo(); local
1023 FunctionDef fdef = test::function::XTimesFour(); local
1070 FunctionDef fdef = test::function::XTimesFour(); local
1221 SetAttrValue(FunctionDef* fdef, const string& attr, const T& value) argument
1229 auto fdef = proto.add_function(); local
1250 auto fdef = proto.add_function(); local
[all...]
H A Dfunction.cc100 // attr_values should specify all attrs defined in fdef.
518 string Print(const FunctionDef& fdef) { argument
520 const OpDef& sig = fdef.signature();
546 for (const auto& n : fdef.node_def()) {
549 for (const auto& r : fdef.ret()) {
627 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values, argument
630 VLOG(3) << "Instantiation Function: " << Print(fdef);
632 const OpDef& sig = fdef.signature();
653 // Makes a copy of all attrs in fdef and substitutes placeholders.
656 node_attrs.resize(fdef
739 GetSetAttrs(const FunctionDef& fdef) argument
774 FunctionDefHash(const FunctionDef& fdef) argument
947 AddFunctionDef(const FunctionDef& fdef) argument
952 AddFunctionDefHelper(const FunctionDef& fdef, bool* added) argument
1155 const FunctionDef* fdef = GetAttrImpl(ndef); local
1218 FunctionDef fdef; local
1248 FunctionDef fdef; local
[all...]
H A Dfunction.h188 // "fdef" encodes a TF function with some attrs in fdef.signature.attr
209 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values,
233 // Return a hash of `fdef` that is consistent with FunctionDefsEqual method.
236 uint64 FunctionDefHash(const FunctionDef& fdef);
301 // Adds function definition 'fdef' to this function library.
303 // 'fdef' already exists in this function library.
304 // If 'fdef' is successfully added to the library, it will be accessible
307 Status AddFunctionDef(const FunctionDef& fdef);
378 FunctionDef fdef; member in struct:tensorflow::FunctionLibraryDefinition::FunctionDefAndOpRegistration
[all...]
/external/tensorflow/tensorflow/c/
H A Dc_api_function.cc163 FunctionDef* fdef) {
167 NodeDef* node_def = fdef->add_node_def();
243 fdef->mutable_signature()->set_is_stateful(true);
257 const char* description, FunctionDef* fdef) {
263 fdef->mutable_signature()->set_description(description);
287 OpDef::ArgDef* argdef = fdef->mutable_signature()->add_output_arg();
301 OpDef::ArgDef* argdef = fdef->mutable_signature()->add_input_arg();
343 FillFunctionBody(fn_name, node_names, body_nodes, tensor_renaming, fdef));
346 for (int r = 0; r < fdef->signature().output_arg_size(); ++r) {
347 const string& ret_name = fdef
159 FillFunctionBody( const string& fn_name, const NodeNameMapping& node_names, const std::vector<const Node*>& body_nodes, const std::unordered_map<string, string>& tensor_renaming, FunctionDef* fdef) argument
251 GraphToFunctionDef(const Graph& fn_body, const string& fn_name, bool append_hash_to_fn_name, const std::vector<const Node*>& body_nodes, const std::vector<OutputTensor>& inputs, const std::vector<OutputTensor>& outputs, const std::vector<string>& output_names, const char* description, FunctionDef* fdef) argument
[all...]
H A Dc_api_function_test.cc226 FunctionDef fdef() { function in class:tensorflow::__anon25329::CApiFunctionTest
227 tensorflow::FunctionDef fdef; local
228 EXPECT_TRUE(GetFunctionDef(func_, &fdef));
229 return fdef;
249 void VerifyFDefNodes(const tensorflow::FunctionDef& fdef, argument
251 ASSERT_EQ(nodes.size(), fdef.node_def_size())
254 << "] Actual nodes in fdef: " << fdef.DebugString();
255 for (const NodeDef& node_def : fdef.node_def()) {
258 << " in fdef
262 VerifyFDefInputs(const tensorflow::FunctionDef& fdef, const std::vector<IOSpec>& inputs) argument
279 VerifyFDefOutputs(const tensorflow::FunctionDef& fdef, const std::vector<IOSpec>& outputs) argument
296 VerifyFDefEdges( const tensorflow::FunctionDef& fdef, const std::vector<EdgeSpec>& e_edges, const std::vector<EdgeSpec>& c_edges, bool is_exact_edges = true) argument
356 tensorflow::FunctionDef fdef; local
367 tensorflow::FunctionDef fdef; local
949 tensorflow::FunctionDef fdef; local
1458 tensorflow::FunctionDef fdef; local
1466 tensorflow::FunctionDef fdef; local
1474 tensorflow::FunctionDef fdef; local
1560 tensorflow::FunctionDef fdef; local
[all...]
H A Dc_api_internal.h161 tensorflow::FunctionDef fdef; member in struct:TF_Function
/external/tensorflow/tensorflow/core/common_runtime/
H A Dfunction.h75 FunctionDef fdef; member in struct:tensorflow::FunctionBody
165 const FunctionDef& fdef, const AttrSlice& attrs,
H A Dfunction.cc228 Status FunctionDefToBody(const FunctionDef& fdef, AttrSlice attrs,
405 &fbody->fdef.signature(), this, fbody->arg_types, input_memory_types,
415 const FunctionDef& fdef, AttrSlice attrs,
418 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig_, fbody);
423 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig, fbody);
430 const FunctionDef* fdef = lib_def->Find(func.name()); local
431 if (fdef == nullptr) {
510 const FunctionDef* fdef = lib_def->Find(function_name); local
511 if (fdef == nullptr) {
514 TF_RETURN_IF_ERROR(FunctionDefToBody(*fdef, attr
414 FunctionDefToBody( const FunctionDef& fdef, AttrSlice attrs, const FunctionLibraryDefinition* lib_def, FunctionBody** fbody) argument
1566 FunctionDefToBodyHelper( const FunctionDef& fdef, const AttrSlice& attrs, const FunctionLibraryDefinition* const lib_def, const std::function<Status(const string&, const OpDef**)>& get_func_sig, FunctionBody** fbody) argument
[all...]
H A Dprocess_function_library_runtime_test.cc42 for (const auto& fdef : flib) *(proto.add_function()) = fdef;
/external/protobuf/ruby/src/main/java/com/google/protobuf/jruby/
H A DRubyMessage.java190 for (Descriptors.FieldDescriptor fdef : descriptor.getFields()) {
191 IRubyObject thisVal = getField(context, fdef);
192 IRubyObject thatVal = message.getField(context, fdef);
352 for (Descriptors.FieldDescriptor fdef : this.descriptor.getFields()) {
353 IRubyObject value = getField(context, fdef);
361 ret.fastASet(runtime.newSymbol(fdef.getName()), value);
403 for (Descriptors.FieldDescriptor fdef : this.descriptor.getFields()) {
404 if (fdef.isRepeated()) {
405 copy.addRepeatedField(fdef, this.getRepeatedField(context, fdef)
[all...]
/external/tensorflow/tensorflow/python/framework/
H A Dfunction.py280 fdef = function_pb2.FunctionDef()
282 fdef.ParseFromString(compat.as_bytes(proto_data))
283 return fdef
795 def _from_definition(fdef, grad_func=None):
799 fdef: a FunctionDef
803 A _DefinedFunction representing fdef
812 argnames = [arg.name for arg in fdef.signature.input_arg]
814 dtypes.as_dtype(arg.type) for arg in fdef.signature.input_arg)
815 func_name = fdef.signature.name
819 out_names = [arg.name for arg in fdef
[all...]
/external/tensorflow/tensorflow/c/eager/
H A Dc_api.cc575 tensorflow::FunctionDef fdef; local
589 tensorflow::OpDef* signature = fdef.mutable_signature();
637 VLOG(1) << "Added OpDef Inputs: " << fdef.DebugString();
650 *fdef.add_node_def() = ndef;
652 fdef.mutable_node_def(0)->set_input(i, op_def.input_arg(i).name());
654 VLOG(1) << "Added NodeDef: " << fdef.DebugString();
663 (*fdef.mutable_ret())[op_def_arg.name()] = out_tensor_name;
676 VLOG(1) << "Fixed Output names and all types: " << fdef.DebugString();
679 status->status = ctx->func_lib_def.AddFunctionDef(fdef);
698 const tensorflow::FunctionDef* fdef; local
[all...]
/external/python/cpython3/Objects/
H A Dmoduleobject.c137 PyMethodDef *fdef; local
139 for (fdef = functions; fdef->ml_name != NULL; fdef++) {
140 if ((fdef->ml_flags & METH_CLASS) ||
141 (fdef->ml_flags & METH_STATIC)) {
147 func = PyCFunction_NewEx(fdef, (PyObject*)module, name);
151 if (PyObject_SetAttrString(module, fdef->ml_name, func) != 0) {
/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcluster_function_library_runtime.cc136 const FunctionDef* fdef = lib_def.Find(function_name); local
137 const OpDef& sig = fdef->signature();
/external/tensorflow/tensorflow/core/grappler/
H A Dutils.cc227 FunctionLibraryDefinition fdef(OpRegistry::Global(), graph->library());
228 auto status = fdef.LookUpOpDef(node.op(), &op_def);
/external/tensorflow/tensorflow/python/eager/
H A Dcontext.py358 def add_function_def(self, fdef):
365 fdef: A FunctionDef protocol buffer message.
367 fdef_string = fdef.SerializeToString()

Completed in 1253 milliseconds

12