Home
last modified time | relevance | path

Searched refs:arg_def (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.cc295 Status AddArgToSig(const NodeDef& node_def, const OpDef::ArgDef& arg_def, in AddArgToSig() argument
298 if (!arg_def.number_attr().empty()) { in AddArgToSig()
301 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, arg_def.number_attr(), &repeats)); in AddArgToSig()
307 if (!arg_def.type_attr().empty()) { in AddArgToSig()
309 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, arg_def.type_attr(), &dtype)); in AddArgToSig()
313 } else if (arg_def.type() != DT_INVALID) { in AddArgToSig()
315 sig->push_back(arg_def.type()); in AddArgToSig()
319 ProtoShortDebugString(arg_def)); in AddArgToSig()
321 } else if (!arg_def.type_attr().empty()) { in AddArgToSig()
324 AttrSlice(node_def).Find(arg_def.type_attr(), &attr_value)); in AddArgToSig()
[all …]
Dfunction.cc50 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def, in ArgNumType() argument
53 if (!arg_def.type_list_attr().empty()) { in ArgNumType()
54 const AttrValue* v = attrs.Find(arg_def.type_list_attr()); in ArgNumType()
57 arg_def.type_list_attr()); in ArgNumType()
68 if (!arg_def.number_attr().empty()) { in ArgNumType()
69 const AttrValue* v = attrs.Find(arg_def.number_attr()); in ArgNumType()
71 return errors::NotFound("type attr not found: ", arg_def.type_attr()); in ArgNumType()
77 if (arg_def.type() != DT_INVALID) { in ArgNumType()
78 dtype = arg_def.type(); in ArgNumType()
79 } else if (arg_def.type_attr().empty()) { in ArgNumType()
[all …]
Dfunction.h132 gtl::ArraySlice<string> arg_def,
138 static FunctionDef Define(gtl::ArraySlice<string> arg_def,
596 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def,
/external/tensorflow/tensorflow/python/framework/
Dimporter.py50 def _ArgToTypesNoRef(node_def, arg_def): argument
51 if arg_def.number_attr:
52 repeats = _GetNodeAttr(node_def, arg_def.number_attr).i
53 if arg_def.type_attr:
54 dtype = _GetNodeAttr(node_def, arg_def.type_attr).type
56 assert arg_def.type != types_pb2.DT_INVALID
57 dtype = arg_def.type
59 elif arg_def.type_attr:
60 return [_GetNodeAttr(node_def, arg_def.type_attr).type]
61 elif arg_def.type_list_attr:
[all …]
Dgraph_to_function_def.py83 for arg_def in op_def.output_arg:
84 if arg_def.number_attr:
85 num = attrs[arg_def.number_attr].i
86 elif arg_def.type_list_attr:
87 num = len(attrs[arg_def.type_list_attr].list.type)
91 result = "%s:%s:%d" % (op.name, arg_def.name, i)
/external/libvpx/libvpx/
Dargs.h24 const struct arg_def *def;
34 typedef struct arg_def { struct
49 int arg_match(struct arg *arg_, const struct arg_def *def, char **argv); argument
51 void arg_show_usage(FILE *fp, const struct arg_def *const *defs);
Dargs.c36 int arg_match(struct arg *arg_, const struct arg_def *def, char **argv) { in arg_match()
90 void arg_show_usage(FILE *fp, const struct arg_def *const *defs) { in arg_show_usage()
94 const struct arg_def *def = *defs; in arg_show_usage()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc170 NodeDef arg_def; in BuildArgNode() local
174 TF_RETURN_IF_ERROR(builder.Finalize(&arg_def)); in BuildArgNode()
175 return AddNode(arg_def, graph); in BuildArgNode()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc663 NodeDef arg_def; in RecordArg() local
669 Status s = builder.Finalize(&arg_def); in RecordArg()
672 Node* arg = graph_->AddNode(arg_def, &s); in RecordArg()