Searched refs:masked_dim (Results 1 – 1 of 1) sorted by relevance
667 int masked_dim, absl::string_view node_name) { in VerifyShapesMatch() argument681 if (dim_i.d[j] != dims_0.d[j] && j != masked_dim) { in VerifyShapesMatch()