Searched refs:arg_def (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | node_def_util.cc | 295 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 …]
|
D | function.cc | 50 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 …]
|
D | function.h | 132 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/ |
D | importer.py | 50 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 …]
|
D | graph_to_function_def.py | 83 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/ |
D | args.h | 24 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);
|
D | args.c | 36 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/ |
D | functionalize_control_flow.cc | 170 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/ |
D | encapsulate_subgraphs_pass.cc | 663 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()
|