Searched refs:dims_to_keep (Results 1 – 1 of 1) sorted by relevance
218 DimensionVector dims_to_keep; in GetReductionKindAndContiguousComponentsImpl() local221 dims_to_keep.push_back(dim); in GetReductionKindAndContiguousComponentsImpl()225 if (dims_to_keep.empty()) { in GetReductionKindAndContiguousComponentsImpl()231 dims_to_keep)) { in GetReductionKindAndContiguousComponentsImpl()233 PartitionShapeByMiddleDimensions(input_shape, dims_to_keep); in GetReductionKindAndContiguousComponentsImpl()266 std::vector<int64> dims_to_keep; in IsReductionFromOrToContiguousDimensions() local269 dims_to_keep.push_back(dim); in IsReductionFromOrToContiguousDimensions()278 dims_to_keep) && in IsReductionFromOrToContiguousDimensions()338 std::vector<int64> dims_to_keep; in IsReductionFromOrToContiguousDimensions() local341 dims_to_keep.push_back(dim); in IsReductionFromOrToContiguousDimensions()[all …]