Searched refs:inferred_attrs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen_internal.cc | 195 const std::unordered_map<string, string>& inferred_attrs, in ArgTypeName() argument
200 gtl::FindOrNull(inferred_attrs, arg.number_attr()); in ArgTypeName()
220 original_arg = gtl::FindOrNull(inferred_attrs, arg.type_attr()); in ArgTypeName()
251 const string* original_arg = gtl::FindOrNull(inferred_attrs, attr_name); in ArgTypeName()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 474 std::unordered_map<string, string>* inferred_attrs) { in InferArgAttributes() argument
476 gtl::InsertIfNotPresent(inferred_attrs, arg.type_attr(), arg.name()); in InferArgAttributes()
478 gtl::InsertIfNotPresent(inferred_attrs, arg.type_list_attr(), arg.name()); in InferArgAttributes()
481 gtl::InsertIfNotPresent(inferred_attrs, arg.number_attr(), arg.name()); in InferArgAttributes()
|