Searched refs:attr_map (Results 1 – 8 of 8) sorted by relevance
57 auto& attr_map = node.attr(); in ExtractExampleParserConfiguration() local58 auto num_sparse = attr_map.at("Nsparse").i(); in ExtractExampleParserConfiguration()59 auto num_dense = attr_map.at("Ndense").i(); in ExtractExampleParserConfiguration()63 auto tdense = attr_map.at("Tdense"); in ExtractExampleParserConfiguration()64 auto dense_shapes = attr_map.at("dense_shapes"); in ExtractExampleParserConfiguration()65 auto sparse_types = attr_map.at("sparse_types"); in ExtractExampleParserConfiguration()
45 attr_map = collections.defaultdict(str, {49 prefix = arch_map[target] + attr_map[target + '-' + attr]
79 auto* attr_map = node->mutable_attr(); in SetNodeAttr() local80 (*attr_map)[key] = attr_value; in SetNodeAttr()
80 const flexbuffers::Map& attr_map = in Initialize() local84 attr_map[kTextEncoderConfigAttr].AsBlob(); in Initialize()
411 auto& attr_map = *d->mutable_attr(); in AddPlaceholdersForFeeds() local412 attr_map["dtype"].set_type(info.data_type); in AddPlaceholdersForFeeds()413 *attr_map["shape"].mutable_shape() = info.feed->shape(); in AddPlaceholdersForFeeds()
382 auto* attr_map = node->mutable_attr(); in SetNodeAttr() local383 (*attr_map)[key] = attr_value; in SetNodeAttr()
488 void FillAttrMap(const OpDef& op_def, AttrMap* attr_map) { in FillAttrMap() argument490 (*attr_map)[attr.name()] = &attr; in FillAttrMap()
4723 def _attr_scope(self, attr_map): argument4752 if not isinstance(attr_map, dict):4759 for name, attr in attr_map.items():4779 for name, attr in attr_map.items():