Home
last modified time | relevance | path

Searched refs:label_mapping (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Deinsum_op_impl.h105 absl::flat_hash_map<char, int>* label_mapping) { in MapToLabels()
113 if (!label_mapping->contains(label_char)) { in MapToLabels()
114 const int next_label = label_mapping->size(); in MapToLabels()
115 (*label_mapping)[label_char] = next_label; in MapToLabels()
117 const int mapped_label = (*label_mapping)[label_char]; in MapToLabels()
140 absl::flat_hash_map<char, int> label_mapping; in ParseEquation() local
146 MapToLabels(input_str[i], &input_labels->at(i), &label_mapping); in ParseEquation()
148 MapToLabels(output_str, output_labels, &label_mapping); in ParseEquation()
151 int num_labels = label_mapping.size(); in ParseEquation()