Searched defs:group_shape (Results 1 – 1 of 1) sorted by relevance
392 ShapeArray* group_shape) { in GroupShapeFromInputs()403 void PopulateGroupIndices(const int64 flat_group_index, VarDimArray group_shape, in PopulateGroupIndices()433 ShapeArray group_shape; in ComputeDenseToDense() local491 ShapeArray group_shape; in ComputeDenseToSparse() local611 ShapeArray group_shape; in ComputeSparseToSparse() local