Searched defs:dims_to_keep (Results 1 – 1 of 1) sorted by relevance
218 DimensionVector dims_to_keep; in GetReductionKindAndContiguousComponentsImpl() local266 std::vector<int64> dims_to_keep; in IsReductionFromOrToContiguousDimensions() local338 std::vector<int64> dims_to_keep; in IsReductionFromOrToContiguousDimensions() local