Searched refs:reshaped_dim (Results 1 – 1 of 1) sorted by relevance
113 void UpdateDDNums(DotDimensionNumbers* new_ddnums, int64 reshaped_dim, in UpdateDDNums() argument116 [&reshaped_dim](tensorflow::protobuf::RepeatedField<int64>* dims) { in UpdateDDNums()119 if (reshaped_dim <= dim) { in UpdateDDNums()123 if (absl::c_linear_search(*dims, reshaped_dim)) { in UpdateDDNums()124 dims->Add(reshaped_dim); in UpdateDDNums()