layout_assignment_test.cc | 405 Shape cshape = ShapeUtil::MakeShape(F32, {4, 3, 2}); local 411 HloInstruction::CreateTranspose(cshape, broadcast, {2, 1, 0})); 417 Shape output_shape_with_layout(cshape); 522 Shape cshape = ShapeUtil::MakeShape(F32, {100}); local 530 HloInstruction::CreateReshape(cshape, concatenate)); [all...] |