/external/tensorflow/tensorflow/compiler/jit/ |
H A D | graph_to_functiondef.h | 25 // Converts 'graph' to a FunctionDef 'fdef', with name 'name'. 29 FunctionDef* fdef);
|
H A D | graph_to_functiondef_test.cc | 63 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 D | graph_to_functiondef.cc | 116 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 D | create_xla_launch_op.cc | 66 DebugString(fbody->fdef)); 94 &fbody->fdef.signature(), flr, fbody->arg_types, input_memory_types,
|
H A D | mark_for_compilation_pass.cc | 126 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 D | encapsulate_subgraphs_pass.cc | 909 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 D | test_util.cc | 26 const FunctionDef* fdef = library.Find(name); local 27 TF_RET_CHECK(fdef != nullptr); 34 InstantiateFunction(*fdef, AttrSlice(), get_func_sig, &inst));
|
H A D | dump_graph.h | 45 string DumpFunctionDefToFile(const string& name, FunctionDef const& fdef);
|
H A D | dump_graph.cc | 85 string DumpFunctionDefToFile(const string& name, FunctionDef const& fdef) { argument 87 Status status = WriteTextProto(Env::Default(), path, fdef);
|
/external/tensorflow/tensorflow/core/framework/ |
H A D | function_test.cc | 67 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 D | function.cc | 100 // 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 D | function.h | 188 // "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 D | c_api_function.cc | 163 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 D | c_api_function_test.cc | 226 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 D | c_api_internal.h | 161 tensorflow::FunctionDef fdef; member in struct:TF_Function
|
/external/tensorflow/tensorflow/core/common_runtime/ |
H A D | function.h | 75 FunctionDef fdef; member in struct:tensorflow::FunctionBody 165 const FunctionDef& fdef, const AttrSlice& attrs,
|
H A D | function.cc | 228 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 D | process_function_library_runtime_test.cc | 42 for (const auto& fdef : flib) *(proto.add_function()) = fdef;
|
/external/protobuf/ruby/src/main/java/com/google/protobuf/jruby/ |
H A D | RubyMessage.java | 190 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 D | function.py | 280 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 D | c_api.cc | 575 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 D | moduleobject.c | 137 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 D | cluster_function_library_runtime.cc | 136 const FunctionDef* fdef = lib_def.Find(function_name); local 137 const OpDef& sig = fdef->signature();
|
/external/tensorflow/tensorflow/core/grappler/ |
H A D | utils.cc | 227 FunctionLibraryDefinition fdef(OpRegistry::Global(), graph->library()); 228 auto status = fdef.LookUpOpDef(node.op(), &op_def);
|
/external/tensorflow/tensorflow/python/eager/ |
H A D | context.py | 358 def add_function_def(self, fdef): 365 fdef: A FunctionDef protocol buffer message. 367 fdef_string = fdef.SerializeToString()
|