Searched refs:attr_map (Results 1 – 7 of 7) 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()
167 auto& attr_map = *d->mutable_attr(); in AddPlaceholdersForFeeds() local168 attr_map["dtype"].set_type(info.data_type); in AddPlaceholdersForFeeds()169 *attr_map["shape"].mutable_shape() = info.feed->shape(); in AddPlaceholdersForFeeds()
79 auto* attr_map = node->mutable_attr(); in SetNodeAttr() local80 (*attr_map)[key] = attr_value; in SetNodeAttr()
45 attr_map = collections.defaultdict(str, {49 prefix = arch_map[target] + attr_map[target + '-' + attr]
373 auto* attr_map = node->mutable_attr(); in SetNodeAttr() local374 (*attr_map)[key] = attr_value; in SetNodeAttr()
474 void FillAttrMap(const OpDef& op_def, AttrMap* attr_map) { in FillAttrMap() argument476 (*attr_map)[attr.name()] = &attr; in FillAttrMap()
4462 def _attr_scope(self, attr_map): argument4491 if not isinstance(attr_map, dict):4498 for name, attr in attr_map.items():4518 for name, attr in attr_map.items():