Home
last modified time | relevance | path

Searched refs:inferred_attrs (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dpython_op_gen_internal.cc195 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/
Dcc_op_gen.cc474 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()