| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph.py | 134 for i, arg_def in enumerate(fdef.signature.input_arg):
136 node_def.name = arg_def.name
138 node_def.attr["dtype"].type = arg_def.type
152 for arg_def in fdef.signature.input_arg:
153 nested_to_flat_tensor_name[arg_def.name] = "{}:0".format(arg_def.name)
154 control_name = "^" + arg_def.name
175 for arg_def in op_def.output_arg:
176 num_args = _get_num_args(arg_def, node_def)
180 nested_name = "{}:{}:{}".format(node_def.name, arg_def.name, i)
196 def _get_num_args(arg_def, node_def): argument
[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/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.cc | 347 Status AddArgToSig(const NodeDef& node_def, const OpDef::ArgDef& arg_def, in AddArgToSig() argument
350 if (!arg_def.number_attr().empty()) { in AddArgToSig()
353 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, arg_def.number_attr(), &repeats)); in AddArgToSig()
359 if (!arg_def.type_attr().empty()) { in AddArgToSig()
361 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, arg_def.type_attr(), &dtype)); in AddArgToSig()
365 } else if (arg_def.type() != DT_INVALID) { in AddArgToSig()
367 sig->push_back(arg_def.type()); in AddArgToSig()
371 ProtoShortDebugString(arg_def)); in AddArgToSig()
373 } else if (!arg_def.type_attr().empty()) { in AddArgToSig()
376 AttrSlice(node_def).Find(arg_def.type_attr(), &attr_value)); in AddArgToSig()
[all …]
|
| D | function.cc | 54 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def, in ArgNumType() argument
57 if (!arg_def.type_list_attr().empty()) { in ArgNumType()
58 const AttrValue* v = attrs.Find(arg_def.type_list_attr()); in ArgNumType()
61 arg_def.type_list_attr()); in ArgNumType()
72 if (!arg_def.number_attr().empty()) { in ArgNumType()
73 const AttrValue* v = attrs.Find(arg_def.number_attr()); in ArgNumType()
75 return errors::NotFound("type attr not found: ", arg_def.type_attr()); in ArgNumType()
81 if (arg_def.type() != DT_INVALID) { in ArgNumType()
82 dtype = arg_def.type(); in ArgNumType()
83 } else if (arg_def.type_attr().empty()) { in ArgNumType()
[all …]
|
| D | function.h | 152 gtl::ArraySlice<string> arg_def,
158 static FunctionDef Define(gtl::ArraySlice<string> arg_def,
754 Status ArgNumType(AttrSlice attrs, const OpDef::ArgDef& arg_def,
|
| /external/tensorflow/tensorflow/java/src/gen/cc/ |
| D | op_specs.cc | 53 Type TypeOf(const OpDef_ArgDef& arg_def, bool* iterable_out);
92 Type TypeResolver::TypeOf(const OpDef_ArgDef& arg_def, bool* iterable_out) { in TypeOf() argument
95 if (arg_def.type() != DataType::DT_INVALID) { in TypeOf()
96 type = Type::ForDataType(arg_def.type()); in TypeOf()
98 } else if (!arg_def.type_attr().empty()) { in TypeOf()
100 if (IsAttributeVisited(arg_def.type_attr())) { in TypeOf()
101 type = visited_attrs_.at(arg_def.type_attr()); in TypeOf()
104 if (attr_def.name() == arg_def.type_attr()) { in TypeOf()
110 } else if (!arg_def.type_list_attr().empty()) { in TypeOf()
114 visited_attrs_.insert(std::make_pair(arg_def.type_list_attr(), type)); in TypeOf()
[all …]
|
| /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/libaom/libaom/common/ |
| D | args.h | 25 const struct arg_def *def;
35 typedef struct arg_def { struct
50 int arg_match(struct arg *arg_, const struct arg_def *def, char **argv); argument
55 void arg_show_usage(FILE *fp, const struct arg_def *const *defs);
|
| D | args.c | 97 int arg_match(struct arg *arg_, const struct arg_def *def, char **argv) { in arg_match()
151 void arg_show_usage(FILE *fp, const struct arg_def *const *defs) { in arg_show_usage()
155 const struct arg_def *def = *defs; in arg_show_usage()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | resource_operation_table_test.cc | 27 bool IsResourceArgDef(const OpDef::ArgDef& arg_def) { in IsResourceArgDef() argument
28 return arg_def.type() == DT_RESOURCE; in IsResourceArgDef()
|
| D | functionalize_while.cc | 136 NodeDef arg_def; in BuildArgNode() local
140 TF_RETURN_IF_ERROR(builder.Finalize(&arg_def)); in BuildArgNode()
141 return AddNodeDefToGraph(arg_def, graph); in BuildArgNode()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 841 OpDef::ArgDef arg_def; in MakeFunctionDef() local
842 arg_def.set_name(input_arg.input_name); in MakeFunctionDef()
843 arg_def.set_type(input_arg.data_type); in MakeFunctionDef()
844 arg_def.set_is_ref(input_arg.is_ref); in MakeFunctionDef()
845 *func->mutable_signature()->add_input_arg() = arg_def; in MakeFunctionDef()
853 OpDef::ArgDef arg_def; in MakeFunctionDef() local
854 arg_def.set_name(output_arg.output_name); in MakeFunctionDef()
855 arg_def.set_type(output_arg.data_type); in MakeFunctionDef()
856 arg_def.set_is_ref(output_arg.is_ref); in MakeFunctionDef()
857 *func->mutable_signature()->add_output_arg() = arg_def; in MakeFunctionDef()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_api_info_test.cc | 32 OpDef::ArgDef* arg_def) { in SetArg() argument
33 arg_def->set_name(name); in SetArg()
34 arg_def->set_type_attr(type_name); in SetArg()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | resource_operation_safety_analysis_test.cc | 518 bool IsResourceArgDef(const OpDef::ArgDef& arg_def) { in IsResourceArgDef() argument
519 return arg_def.type() == DT_RESOURCE; in IsResourceArgDef()
|
| D | extract_outside_compilation_pass.cc | 860 NodeDef arg_def; in ReplaceKeyPlaceholderWithArgNode() local
861 TF_RETURN_IF_ERROR(arg_builder.Finalize(&arg_def)); in ReplaceKeyPlaceholderWithArgNode()
862 TF_RETURN_IF_ERROR(ReplaceNode(g, key_placeholder, arg_def).status()); in ReplaceKeyPlaceholderWithArgNode()
|
| D | encapsulate_subgraphs_pass.cc | 781 NodeDef arg_def; in RecordArg() local
788 Status s = builder.Finalize(&arg_def); in RecordArg()
791 Node* arg = graph_->AddNode(arg_def, &s); in RecordArg()
|