Searched refs:ForEachMutableSubshape (Results 1 – 18 of 18) sorted by relevance
111 ShapeUtil::ForEachMutableSubshape(&result_layout, in CompileAheadOfTime()116 ShapeUtil::ForEachMutableSubshape(const_cast<Shape*>(shape), in CompileAheadOfTime()
155 ShapeUtil::ForEachMutableSubshape( in UpdateEntryComputationLayout()
176 ShapeUtil::ForEachMutableSubshape( in ChangeOutputTypeThenInsertConvertBack()
649 ShapeUtil::ForEachMutableSubshape( in ExecuteGraphParallel()
2066 xla::ShapeUtil::ForEachMutableSubshape( in AssignLayouts()
2956 ShapeUtil::ForEachMutableSubshape( in InferConditionalShape()
331 ShapeUtil::ForEachMutableSubshape( in ComputeAndCompareLiteralWithStatus()389 ShapeUtil::ForEachMutableSubshape( in ComputeAndCompareLiteralWithStatus()623 ShapeUtil::ForEachMutableSubshape(&new_shape, in MaybeConvertShapeToBfloat16()
75 xla::ShapeUtil::ForEachMutableSubshape(&device_shape, update_layout); in AllocateOutputMemoryWithInputReuse()
429 ShapeUtil::ForEachMutableSubshape( in AddOutfeedToBuilder()
162 ShapeUtil::ForEachMutableSubshape( in PYBIND11_MODULE()
344 ShapeUtil::ForEachMutableSubshape( in BuildXlaCompilerSubmodule()
286 ShapeUtil::ForEachMutableSubshape( in ClearTiles()
477 static void ForEachMutableSubshape(Shape* shape,
49 ShapeUtil::ForEachMutableSubshape( in ConvertType()
1111 /* static */ void ShapeUtil::ForEachMutableSubshape( in ForEachMutableSubshape() function in xla::ShapeUtil1729 ForEachMutableSubshape(&s, [](Shape* subshape, const ShapeIndex& index) { in DeviceShapeToHostShape()
554 ShapeUtil::ForEachMutableSubshape( in TEST()
899 ShapeUtil::ForEachMutableSubshape( in ToStatic()
4547 ShapeUtil::ForEachMutableSubshape(new_program_shape.mutable_parameters(i), in Run()4552 ShapeUtil::ForEachMutableSubshape(new_program_shape.mutable_result(), in Run()