Searched refs:ForEachMutableSubshape (Results 1 – 7 of 7) sorted by relevance
331 ShapeUtil::ForEachMutableSubshape( in ComputeAndCompareLiteralWithStatus()388 ShapeUtil::ForEachMutableSubshape( in ComputeAndCompareLiteralWithStatus()623 ShapeUtil::ForEachMutableSubshape(&new_shape, in MaybeConvertShapeToBfloat16()
177 ShapeUtil::ForEachMutableSubshape( in ChangeOutputTypeThenInsertConvertBack()
50 ShapeUtil::ForEachMutableSubshape( in ConvertType()
529 static void ForEachMutableSubshape(Shape* shape,
537 ShapeUtil::ForEachMutableSubshape( in TEST()
905 /* static */ void ShapeUtil::ForEachMutableSubshape( in ForEachMutableSubshape() function in xla::ShapeUtil
416 ShapeUtil::ForEachMutableSubshape( in PYBIND11_MODULE()