Searched refs:attr_slice (Results 1 – 4 of 4) sorted by relevance
125 for (const auto& attr_slice : node.attrs()) { in CwiseVectorizeHelper() local126 node_builder = node_builder.Attr(attr_slice.first, attr_slice.second); in CwiseVectorizeHelper()
471 const AttrSlice attr_slice = in TEST() local476 EXPECT_TRUE(OutputTypesForNode(attr_slice, op_def, &types).ok()); in TEST()
695 AttrSlice attr_slice(attrs); in AddDefaultAttrs() local697 if (attr_def.has_default_value() && !attr_slice.Find(attr_def.name())) { in AddDefaultAttrs()
809 AttrSlice attr_slice = AttrSlice(&remote_op.attrs()); in StoreResourceDtypesAndShapes() local811 TF_RETURN_IF_ERROR(attr_slice.Find("dtype", &dtype)); in StoreResourceDtypesAndShapes()813 TF_RETURN_IF_ERROR(attr_slice.Find("shape", &shape)); in StoreResourceDtypesAndShapes()