Searched refs:ForEachMutableSubshape (Results 1 – 5 of 5) sorted by relevance
331 ShapeUtil::ForEachMutableSubshape( in ComputeAndCompareLiteralWithStatus()388 ShapeUtil::ForEachMutableSubshape( in ComputeAndCompareLiteralWithStatus()623 ShapeUtil::ForEachMutableSubshape(&new_shape, in MaybeConvertShapeToBfloat16()
50 ShapeUtil::ForEachMutableSubshape( in ConvertType()
516 static void ForEachMutableSubshape(Shape* shape,
515 ShapeUtil::ForEachMutableSubshape( in TEST()
905 /* static */ void ShapeUtil::ForEachMutableSubshape( in ForEachMutableSubshape() function in xla::ShapeUtil