Searched refs:inferred_attrs_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen_internal.cc | 538 gtl::InsertIfNotPresent(&inferred_attrs_, arg.type_attr(), arg.name()); in Code()
540 gtl::InsertIfNotPresent(&inferred_attrs_, arg.type_list_attr(), in Code()
544 gtl::InsertIfNotPresent(&inferred_attrs_, arg.number_attr(), arg.name()); in Code()
550 if (inferred_attrs_.find(attr.name()) == inferred_attrs_.end()) { in Code()
697 ArgTypeName(op_def_, arg, inferred_attrs_, false)); in AddDocStringInputs()
815 ArgTypeName(op_def_, op_def_.output_arg(i), inferred_attrs_, true)); in AddDocStringOutputs()
|
| D | python_op_gen.cc | 189 gtl::InsertIfNotPresent(&inferred_attrs_, attr, in AddAttrForArg()
296 if (inferred_attrs_.find(attr.name()) == inferred_attrs_.end()) { in Code()
673 inferred_attrs_[attr.name()], "'.\" %\n", indentation, in GetEagerFunctionSetup()
789 const auto iter = inferred_attrs_.find(arg.type_list_attr()); in GetOutputSizesAndNumOutputsExpr()
790 if (iter == inferred_attrs_.end()) { in GetOutputSizesAndNumOutputsExpr()
942 if (inferred_attrs_.find(attr.name()) == inferred_attrs_.end()) { in AddEagerFastPathExecute()
|
| D | python_op_gen_internal.h | 109 std::unordered_map<string, string> inferred_attrs_; variable
|