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()
|