Searched refs:inferred_attrs (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_api_parameter_converter.cc | 250 InferredAttributes* inferred_attrs) { in ConvertInputsWithTypeAttr() argument
299 if (inferred_attrs) { in ConvertInputsWithTypeAttr()
308 inferred_attrs->types[inferred_index] = dtype; in ConvertInputsWithTypeAttr()
325 InferredAttributes* inferred_attrs) { in ConvertInputsWithTypeListAttr() argument
385 if (inferred_attrs) { in ConvertInputsWithTypeListAttr()
388 inferred_attrs->type_lists[inferred_index].swap(dtypes); in ConvertInputsWithTypeListAttr()
438 InferredAttributes* inferred_attrs) { in ConvertPythonAPIParameters() argument
440 if (inferred_attrs) { in ConvertPythonAPIParameters()
441 inferred_attrs->types.resize(api_info.inferred_type_attrs().size()); in ConvertPythonAPIParameters()
442 inferred_attrs->type_lists.resize( in ConvertPythonAPIParameters()
[all …]
|
| D | python_api_parameter_converter_wrapper.cc | 31 PythonAPIInfo::InferredAttributes inferred_attrs; in Convert() local
41 &inferred_attrs)) { in Convert()
45 return inferred_attrs; in Convert()
|
| D | python_api_info.cc | 72 std::set<std::string> inferred_attrs; in GetOpDefNamesAndDefaults() local
78 inferred_attrs.insert(input.type_attr()); in GetOpDefNamesAndDefaults()
81 inferred_attrs.insert(input.type_list_attr()); in GetOpDefNamesAndDefaults()
84 inferred_attrs.insert(input.number_attr()); in GetOpDefNamesAndDefaults()
93 if (inferred_attrs.count(attr.name()) == 0) { in GetOpDefNamesAndDefaults()
|
| D | python_api_parameter_converter.h | 58 PythonAPIInfo::InferredAttributes* inferred_attrs);
|
| D | python_op_gen_internal.cc | 199 const std::unordered_map<string, string>& inferred_attrs, in ArgTypeName() argument
204 gtl::FindOrNull(inferred_attrs, arg.number_attr()); in ArgTypeName()
224 original_arg = gtl::FindOrNull(inferred_attrs, arg.type_attr()); in ArgTypeName()
255 const string* original_arg = gtl::FindOrNull(inferred_attrs, attr_name); in ArgTypeName()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 496 std::unordered_map<string, string>* inferred_attrs) { in InferArgAttributes() argument
498 gtl::InsertIfNotPresent(inferred_attrs, arg.type_attr(), arg.name()); in InferArgAttributes()
500 gtl::InsertIfNotPresent(inferred_attrs, arg.type_list_attr(), arg.name()); in InferArgAttributes()
503 gtl::InsertIfNotPresent(inferred_attrs, arg.number_attr(), arg.name()); in InferArgAttributes()
|