Searched refs:node_attr (Results 1 – 5 of 5) sorted by relevance
70 Attr* node_attr = absl::any_cast<Attr>(&op_node->operation.attributes); in ApplyToNodesSequence() local78 node_attr->padding.appended.h += pad_attr.appended.h; in ApplyToNodesSequence()79 node_attr->padding.appended.w += pad_attr.appended.w; in ApplyToNodesSequence()80 node_attr->padding.prepended.h += pad_attr.prepended.h; in ApplyToNodesSequence()81 node_attr->padding.prepended.w += pad_attr.prepended.w; in ApplyToNodesSequence()
47 auto& node_attr = in ApplyToNode() local49 auto& node_code = node_attr.code; in ApplyToNode()212 if (!MergeCode(&attr, &node_attr).ok()) { in ApplyToNode()215 absl::StrAppend(&node_attr.code.source_code, "{\n", attr.code.source_code, in ApplyToNode()
274 for (const auto& node_attr : node->op_def().attr()) { in FillFunctionBody() local275 if (node_attr.name() == node_attr_name) { in FillFunctionBody()276 node_attr_def = &node_attr; in FillFunctionBody()
679 for (const auto& node_attr : attrs) { in GetFunctionNameAttr() local680 if (node_attr.first == group_attribute_) { in GetFunctionNameAttr()681 TF_RETURN_IF_ERROR(AttrValueHasType(node_attr.second, "string")); in GetFunctionNameAttr()682 *attr = node_attr.second.s(); in GetFunctionNameAttr()
1491 for (const auto& node_attr : node->attrs()) { in ConvertFunctionArgAndRets() local1492 const auto& key = node_attr.first; in ConvertFunctionArgAndRets()1498 if (IsOutputShapesAttribute(node_attr.second, key) || in ConvertFunctionArgAndRets()1499 IsResourceOutputShapesAttribute(node_attr.second, key)) in ConvertFunctionArgAndRets()1502 ConvertAttributeValue(node_attr.second)); in ConvertFunctionArgAndRets()