Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc208 absl::flat_hash_map<int64, int64> result_dim_mapping; in HandleDot() local
215 result_dim_mapping[i] = current_result_dims++; in HandleDot()
222 result_dim_mapping[i] = current_result_dims; in HandleDot()
234 result_dim_mapping[i] = current_result_dims; in HandleDot()
242 auto iter = result_dim_mapping.find(dimension); in HandleDot()
243 if (iter != result_dim_mapping.end()) { in HandleDot()