Searched refs:non_contracting_non_batch_dim (Results 1 – 1 of 1) sorted by relevance
1003 absl::optional<int64> non_contracting_non_batch_dim = in CanFoldDotIntoIndexedArray() local1006 if (!non_contracting_non_batch_dim.has_value()) { in CanFoldDotIntoIndexedArray()1012 indexed_array->output_dims()[0] != *non_contracting_non_batch_dim) { in CanFoldDotIntoIndexedArray()