Searched refs:ClearLayout (Results 1 – 9 of 9) sorted by relevance
259 /* static */ void LayoutUtil::ClearLayout(Shape* shape) { in ClearLayout() function in xla::LayoutUtil262 ClearLayout(&element_shape); in ClearLayout()266 /* static */ void LayoutUtil::ClearLayout(ProgramShape* program_shape) { in ClearLayout() function in xla::LayoutUtil268 LayoutUtil::ClearLayout(¶meter_shape); in ClearLayout()270 LayoutUtil::ClearLayout(program_shape->mutable_result()); in ClearLayout()
88 static void ClearLayout(Shape* shape);91 static void ClearLayout(ProgramShape* program_shape);
273 LayoutUtil::ClearLayout(&shape); in TEST_F()285 LayoutUtil::ClearLayout(&shape); in TEST_F()424 LayoutUtil::ClearLayout(&shape); in TEST_F()454 LayoutUtil::ClearLayout(&sub_1_2); in TEST_F()457 LayoutUtil::ClearLayout(&sub_2_1); in TEST_F()
59 void ShapeLayout::Clear() { LayoutUtil::ClearLayout(&shape_); } in Clear()
66 LayoutUtil::ClearLayout(&shape); in CombineConstants()
807 LayoutUtil::ClearLayout(tuple_copy->mutable_shape()); in CreateCopyWithNewLayout()817 LayoutUtil::ClearLayout(copy->mutable_shape()); in CreateCopyWithNewLayout()1574 LayoutUtil::ClearLayout(fused_instruction->mutable_shape()); in SetFusionLayouts()1590 LayoutUtil::ClearLayout(instruction->mutable_shape()); in AssignLayouts()1693 LayoutUtil::ClearLayout(instruction->mutable_shape()); in ClearComputationLayouts()2135 LayoutUtil::ClearLayout(copy->mutable_shape()); in AddCopyForOperand()
242 LayoutUtil::ClearLayout(add->mutable_shape()); in TEST_F()
629 LayoutUtil::ClearLayout(hlo->mutable_shape()); in TEST_F()646 LayoutUtil::ClearLayout(hlo->mutable_shape()); in TEST_F()
803 xla::LayoutUtil::ClearLayout(&input_shape); in TEST()804 xla::LayoutUtil::ClearLayout(&kernel_shape); in TEST()816 xla::LayoutUtil::ClearLayout(&result_shape); in TEST()