Searched refs:padded_result_buffer_shape (Results 1 – 1 of 1) sorted by relevance
604 auto padded_result_buffer_shape = unpadded_result_buffer_shape; in PartitionBaseCase() local612 padded_result_buffer_shape = GetPaddedShapeForUnevenPartitioning( in PartitionBaseCase()613 padded_result_buffer_shape, in PartitionBaseCase()631 auto result_buffer = CreateZero(padded_result_buffer_shape, b); in PartitionBaseCase()635 ? CreateZero(padded_result_buffer_shape, b) in PartitionBaseCase()1457 if (!ShapeUtil::Compatible(padded_result_buffer_shape, in PartitionBaseCase()1461 std::vector<int64>(padded_result_buffer_shape.rank(), 0), in PartitionBaseCase()1463 std::vector<int64>(padded_result_buffer_shape.rank(), 1))); in PartitionBaseCase()