Searched refs:node_attrs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 1289 const string& GetKernelLabelAttr(const AttrSlice& node_attrs) { in GetKernelLabelAttr() argument
1297 const AttrValue* attr_value = node_attrs.FindByString(kKernelAttr); in GetKernelLabelAttr()
1309 StringPiece node_op, AttrSlice node_attrs, const KernelRegistration** reg, in FindKernelRegistration() argument
1314 const string& label = GetKernelLabelAttr(node_attrs); in FindKernelRegistration()
1324 TF_RETURN_IF_ERROR(KernelAttrsMatch(iter->second.def, node_attrs, &match)); in FindKernelRegistration()
1356 KernelAttrsMatch(iter->second.def, node_attrs, &match)); in FindKernelRegistration()
1410 StringPiece node_op, StringPiece node_device, AttrSlice node_attrs, in FindKernelDef() argument
1416 experimental_debug_info, node_op, node_attrs, ®, &was_attr_mismatch)); in FindKernelDef()
1428 SummarizeAttrsHelper(node_attrs, node_device)); in FindKernelDef()
|
| D | function.cc | 761 std::vector<AttrValueMap> node_attrs; in InstantiateFunction() local
762 node_attrs.resize(fdef.node_def_size()); in InstantiateFunction()
769 if (!node_attrs[i].insert(attr).second) { in InstantiateFunction()
774 AddDefaultAttrs(fdef.node_def(i).op(), get_function, &node_attrs[i])); in InstantiateFunction()
778 s = helper.BuildNodeOutputIndex(fdef.node_def(i), AttrSlice(&node_attrs[i]), in InstantiateFunction()
788 s = helper.InstantiateNode(fdef.node_def(i), AttrSlice(&node_attrs[i])); in InstantiateFunction()
|
| D | op_kernel.h | 1502 StringPiece node_op, StringPiece node_device, AttrSlice node_attrs,
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 496 AttrSlice node_attrs) { in IsKernelRegisteredForNode() argument
504 node_op, node_device, node_attrs, nullptr, nullptr); in IsKernelRegisteredForNode()
|
| D | utils.h | 395 AttrSlice node_attrs);
|