Searched refs:label_to_dimension (Results 1 – 1 of 1) sorted by relevance
179 absl::flat_hash_map<char, DimensionHandle> label_to_dimension; in EinsumShape() local240 if (label_to_dimension.contains(label)) { in EinsumShape()243 c->Merge(label_to_dimension[label], new_dim, &merged)); in EinsumShape()244 label_to_dimension[label] = merged; in EinsumShape()246 label_to_dimension[label] = new_dim; in EinsumShape()297 auto dimension_it = label_to_dimension.find(label); in EinsumShape()298 if (dimension_it == label_to_dimension.end()) { in EinsumShape()