Searched defs:arg_attr (Results 1 – 7 of 7) sorted by relevance
41 map<uint32, ArgAttrs> arg_attr = 7; field
207 for (const auto& arg_attr : arg_attrs->attr()) { in BuildInputArgIndex() local1006 auto arg_attr = iter->second.attr().find("_composite_device"); in GetFunctionResourceInputDevice() local
41 std::vector<const FunctionDef::ArgAttrs*> arg_attr, in GrapplerFunctionItem()114 GrapplerFunctionItem::arg_attr() const { in arg_attr() function in tensorflow::grappler::GrapplerFunctionItem292 std::vector<const FunctionDef::ArgAttrs*> arg_attr(inputs.size(), nullptr); in MakeGrapplerFunctionItem() local
529 FunctionDef::ArgAttrs arg_attr; in TEST_F() local
578 auto arg_attr = iter->second.attr().find("_composite_device"); in Instantiate() local
1587 for (const auto& arg_attr : dict_attrs) { in GetStringsFromDictionaryAttr() local
669 for (auto &arg_attr : arg_attrs) { in matchAndRewrite() local