Searched refs:node_attrs (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.cc | 1354 const string& GetKernelLabelAttr(const AttrSlice& node_attrs) { in GetKernelLabelAttr() argument
1362 const AttrValue* attr_value = node_attrs.FindByString(kKernelAttr); in GetKernelLabelAttr()
1374 StringPiece node_op, AttrSlice node_attrs, const KernelRegistration** reg, in FindKernelRegistration() argument
1379 const string& label = GetKernelLabelAttr(node_attrs); in FindKernelRegistration()
1389 TF_RETURN_IF_ERROR(KernelAttrsMatch(iter->second.def, node_attrs, &match)); in FindKernelRegistration()
1421 KernelAttrsMatch(iter->second.def, node_attrs, &match)); in FindKernelRegistration()
1475 StringPiece node_op, StringPiece node_device, AttrSlice node_attrs, in FindKernelDef() argument
1481 experimental_debug_info, node_op, node_attrs, ®, &was_attr_mismatch)); in FindKernelDef()
1493 SummarizeAttrsHelper(node_attrs, node_device)); in FindKernelDef()
|
| H A D | function.cc | 802 std::vector<AttrValueMap> node_attrs; in InstantiateFunction() local
803 node_attrs.resize(fdef.node_def_size()); in InstantiateFunction()
810 if (!node_attrs[i].insert(attr).second) { in InstantiateFunction()
815 AddDefaultAttrs(fdef.node_def(i).op(), get_function, &node_attrs[i])); in InstantiateFunction()
819 s = helper.BuildNodeOutputIndex(fdef.node_def(i), AttrSlice(&node_attrs[i]), in InstantiateFunction()
829 s = helper.InstantiateNode(fdef.node_def(i), AttrSlice(&node_attrs[i])); in InstantiateFunction()
|
| H A D | op_kernel.h | 1471 StringPiece node_op, StringPiece node_device, AttrSlice node_attrs,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | utils.cc | 481 AttrSlice node_attrs) { in IsKernelRegisteredForNode() argument
489 node_op, node_device, node_attrs, nullptr, nullptr); in IsKernelRegisteredForNode()
|
| H A D | utils.h | 409 AttrSlice node_attrs);
|